Find Right Interval

题目来源
给一堆intervals,然后让你求每一个interval右边的最小的interval。
我的做法是先排个序,然后搞个map来映射索引和start,因为题目说start是唯一的。然后二分搜索来确定这个索引。

/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
class Solution {
public:
    vector<int> findRightInterval(vector<Interval>& intervals) {
        map<int, int> maps;
        vector<Interval> intervals_copy(intervals);
        int n = intervals.size();
        for (int i=0; i<n; i++)
            maps[intervals[i].start] = i;
        auto cmp = [](const Interval& i1, const Interval& i2){ return i1.start < i2.start; };
        sort(intervals_copy.begin(), intervals_copy.end(), cmp);
        vector<int> res(n, -1);
        for (int i=0; i<n; i++) {
            int target = intervals[i].end;
            int l = 0, r = n - 1, mid = 0;
            if (intervals_copy[r].start < target)
                continue;
            while (l < r) {
                mid = (l + r) / 2;
                if (intervals_copy[mid].start == target) {
                    r = mid;
                    break;
                }
                else if (intervals_copy[mid].start < target)
                    l = mid + 1;
                else
                    r = mid;
            }
            res[i] = maps[intervals_copy[r].start];
        }
        return res;
    }
};

看了看讨论区,发现自己有点傻,不对,主要是STL的各种还不熟悉,不会利用,用了之后就特别简单简洁,代码如下:

class Solution {
public:
    vector<int> findRightInterval(vector<Interval>& intervals) {
        map<int, int> hash;
        vector<int> res;
        int n = intervals.size();
        for (int i = 0; i < n; ++i)
            hash[intervals[i].start] = i;
        for (auto in : intervals) {
            auto itr = hash.lower_bound(in.end);
            if (itr == hash.end()) res.push_back(-1);
            else res.push_back(itr->second);
        }
        return res;
    }
};
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Difficulty: MediumProblem Link: https://leetcode.com/prob...
    Leopzm阅读 668评论 0 0
  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,789评论 0 33
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 135,107评论 19 139
  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 17,984评论 2 36
  • ——《一本杂志和一个时代的体温——<新周刊>20年精选》书评 《新周刊》是许多人很喜欢的一本杂志,喜欢它犀利的批判...
    山千黛阅读 242评论 0 2