leetcode 436. 寻找右区间

题目链接：leetcode 436

1.题目

给你一个区间数组 intervals ，其中 intervals[i] = [starti, endi] ，且每个 starti 都 不同 。

区间 i 的 右侧区间 可以记作区间 j ，并满足 startj >= endi ，且 startj 最小化 。注意 i 可能等于 j 。

返回一个由每个区间 i 的 右侧区间 在 intervals 中对应下标组成的数组。如果某个区间 i 不存在对应的 右侧区间 ，则下标 i 处的值设为 -1 。

2.示例

1）示例 1：
输入：intervals = [[1,2]]
输出：[-1]
解释：集合中只有一个区间，所以输出-1。

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

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

4）提示：
1 <= intervals.length <= 2 * 104
intervals[i].length == 2
-106 <= starti <= endi <= 106
每个间隔的起点都 不相同

3.分析

可以发现对于一个确定的右边界点j，可以使用二分的方法对已经排列好的左边界点数组进行查找

4.代码

class Solution {
public:
    map<int,int> map1;
    vector<int> nums,ans;
    vector<int> findRightInterval(vector<vector<int>>& intervals) {
        for(int i=0;i<intervals.size();i++){
            map1[intervals[i][0]]=i;
            nums.push_back(intervals[i][0]);
        }
        sort(nums.begin(),nums.end());
        for(int i=0;i<intervals.size();i++){
            int ed=intervals[i][1];
            int left=0,right=nums.size()-1;
            while(left+1<right){
                int mid=(left+right)/2;
                if(nums[mid]<ed) left=mid;
                else right=mid;
            }
            if(nums[left]>=ed) ans.push_back(map1[nums[left]]);
            else{
                if(nums[right]>=ed) ans.push_back(map1[nums[right]]);
                else ans.push_back(-1);
            }
        }
        return ans;
    }
};