[C++] LeetCode 436. 寻找右区间

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/lv1224/article/details/81751189

题目

给定一组区间，对于每一个区间 i，检查是否存在一个区间 j，它的起始点大于或等于区间 i 的终点，这可以称为 j 在 i 的“右侧”。

对于任何区间，你需要存储的满足条件的区间 j 的最小索引，这意味着区间 j 有最小的起始点可以使其成为“右侧”区间。如果区间 j 不存在，则将区间 i 存储为 -1。最后，你需要输出一个值为存储的区间值的数组。

注意:
你可以假设区间的终点总是大于它的起始点。
你可以假定这些区间都不具有相同的起始点。

示例 1:

输入: [ [1,2] ]
输出: [-1]
解释:集合中只有一个区间，所以输出-1。

示例 2:

输入: [ [3,4], [2,3], [1,2] ]
输出: [-1, 0, 1]
解释:对于[3,4]，没有满足条件的“右侧”区间。
对于[2,3]，区间[3,4]具有最小的“右”起点;
对于[1,2]，区间[2,3]具有最小的“右”起点。

示例 3:

输入: [ [1,4], [2,3], [3,4] ]
输出: [-1, 2, -1]
解释:对于区间[1,4]和[3,4]，没有满足条件的“右侧”区间。
对于[2,3]，区间[3,4]有最小的“右”起点。

题解

这道题目考虑用二分法求解。用一个数组维护所有区间的起始点，那么对于一个区间，它的右区间只需要找出大于或等于它的右边界的值即可，那么对于起始点构成的数组如果有序则可以直接二分查找。

代码一

/**
 * 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) {
        int n=intervals.size();
        unordered_map<int,int> m;
        vector<int> vec(n,0),res(n,-1);
        for(int i=0;i<n;i++){
            int tmp=intervals[i].start;
            vec[i]=intervals[i].start;
            m[tmp]=i;
        }
        sort(vec.begin(),vec.end());
        for(int i=0;i<n;i++){
            int end=intervals[i].end;
            auto it=lower_bound(vec.begin(),vec.end(),end);
            if(it!=vec.end()) res[i]=m[*it]; 
        }
        return res;
    }
};

代码2

/**
 * 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:
    static bool cmp(pair<int,int> &a,pair<int,int> &b){
        return a.first<b.first;
    }
    static bool cmp2(pair<int,int> &a,int b){
        return a.first<b;
    }
    vector<int> findRightInterval(vector<Interval>& intervals) {
        int n=intervals.size();
        vector<pair<int,int>> vec(n);
        vector<int> res(n,-1);
        for(int i=0;i<n;i++){
            int tmp=intervals[i].start;
            vec[i]=make_pair(tmp,i);
        }
        sort(vec.begin(),vec.end(),cmp);
        for(int i=0;i<n;i++){
            int end=intervals[i].end;
            auto it=lower_bound(vec.begin(),vec.end(),end,cmp2);
            if(it!=vec.end()) res[i]=(*it).second; 
        }
        return res;
    }
};