原题链接: 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
每个间隔的起点都 不相同
思路
排序、二分
- 将所有区间按照左端点排序
- 遍历区间,对每个区间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).