美文网首页
Find Right Interval

Find Right Interval

作者: 我叫胆小我喜欢小心 | 来源:发表于2017-05-25 11:02 被阅读20次

    题目来源
    给一堆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;
        }
    };
    

    相关文章

      网友评论

          本文标题:Find Right Interval

          本文链接:https://www.haomeiwen.com/subject/wnwaxxtx.html