【LeetCode】436. 寻找右区间

原题链接： 436. 寻找右区间

题目大意

给你一个区间数组 intervals ，其中 intervals[i] = [starti, endi] ，且每个starti都 不同 。
区间 i 的 右侧区间 可以记作区间j，并满足startj >= endi，且startj最小化 。
返回一个由每个区间 i 的 右侧区间 的最小起始位置组成的数组。如果某个区间i不存在对应的 右侧区间 ，则下标i处的值设为 -1 。
示例：

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

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

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

数据范围：

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

思路

排序、二分

1. 将所有区间按照左端点排序
2. 遍历区间，对每个区间i，二分查找区间j，使得startj>=endi，且startj最小。

代码

class Solution {
public:
    vector<int> findRightInterval(vector<vector<int>>& q) {
        int n = q.size();
        for(int i = 0; i < n; i ++ ) q[i].push_back(i); // 存位置
        sort(q.begin(), q.end());

        vector<int> res(n, -1);
        for(auto x : q){
            int l = 0, r = n - 1;
            while(l < r){
                int mid = l + r >> 1;
                if(q[mid][0] >= x[1]) r = mid;
                else l = mid + 1;
            }
            if(q[l][0] >= x[1]) res[x[2]] = q[l][2];
        }

        return res;
    }
};

复杂度

时间复杂度：排序O(nlogn)，二分O(nlogn)，故总时间复杂度为O(nlogn).
空间复杂度：需要答案数组，以及存储位置的空间，故空间复杂度为O(n).