题目大意
给定一组区间,对于每一个区间 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]具有最小的“右”起点。
解题思路
这道题求的是,对于每一个区间(x,y),在区间数组中找到第一个大于等于y(即大于等于y,且最接近y)的区间的索引。
很容易想到另一个题目:给定有序数组和target,找到第一个大于等于target的数字的索引。
因此我们将原区间数组按照区间端点升序排列,同时记录该区间在原排列中的索引。
因此,对于任意一个区间(x,y),y等价于target,所有区间的x等价于有序数组,在其中二分查找即可。
class Solution {
private:
int findLower(vector<pair<int, int>> & nums, int target){
int left = 0, right = nums.size() - 1, mid = 0;
while (left <= right){
mid = left + (right - left) / 2;
if (nums[mid].first < target)
left = mid + 1;
else
right = mid - 1;
}
return left == nums.size() ? -1 : nums[left].second;
}
public:
vector<int> findRightInterval(vector<vector<int>>& intervals) {
if (intervals.size() == 1)
return {-1};
vector<pair<int, int>> nums;
for (int i = 0; i < intervals.size(); ++i){
nums.push_back({intervals[i][0], i});
}
sort(nums.begin(), nums.end());
vector<int> res(intervals.size(), -1);
int tmp = -1, target = 0;
for(int i = 0; i < intervals.size(); ++i){
target = intervals[i][1];
tmp = findLower(nums, target);
res[i] = tmp;
}
return res;
}
};