给你一个区间数组 intervals ,其中 intervals[i] = [starti, endi] ,且每个 starti 都 不同 。
区间 i 的 右侧区间 可以记作区间 j ,并满足 startj >= endi ,且 startj 最小化 。
返回一个由每个区间 i 的 右侧区间 在 intervals 中对应下标组成的数组。如果某个区间 i 不存在对应的 右侧区间 ,则下标 i 处的值设为 -1 。
示例 1:
输入:intervals = [[1,2]]
输出:[-1]
解释:集合中只有一个区间,所以输出-1。
示例 2:
输入:intervals = [[3,4],[2,3],[1,2]]
输出:[-1,0,1]
解释:对于 [3,4] ,没有满足条件的“右侧”区间。
对于 [2,3] ,区间[3,4]具有最小的“右”起点;
对于 [1,2] ,区间[2,3]具有最小的“右”起点。
示例 3:
输入:intervals = [[1,4],[2,3],[3,4]]
输出:[-1,2,-1]
解释:对于区间 [1,4] 和 [3,4] ,没有满足条件的“右侧”区间。
对于 [2,3] ,区间 [3,4] 有最小的“右”起点。
二分查找
class Solution {
public int[] findRightInterval(int[][] intervals) {
int n = intervals.length;
int[][] start = new int[n][2];
for(int i = 0; i < n; i++) {
start[i][0] = intervals[i][0];
start[i][1] = i;
}
Arrays.sort(start, (o1, o2) -> o1[0] - o2[0]);
int[] res = new int[n];
for(int i = 0; i < n; i++) {
int left = 0, right = n - 1;
int target = -1;
while(left <= right) {
int mid = left + (right - left) / 2;
if(intervals[i][1] <= start[mid][0]) {
target = start[mid][1];
right = mid - 1;
} else {
left = mid + 1;
}
}
res[i] = target;
}
return res;
}
}
双指针
class Solution {
public int[] findRightInterval(int[][] intervals) {
int n = intervals.length;
int[][] start = new int[n][2];
int[][] end = new int[n][2];
for(int i = 0; i < n; i++) {
start[i][0] = intervals[i][0];
start[i][1] = i;
end[i][0] = intervals[i][1];
end[i][1] = i;
}
Arrays.sort(start, (o1, o2) -> o1[0] - o2[0]);
Arrays.sort(end, (o1, o2) -> o1[0] - o2[0]);
int[] res = new int[n];
for(int i = 0, j = 0; i < n; i++) {
while(j < n && end[i][0] > start[j][0]) {
j++;
}
if(j < n) {
res[end[i][1]] = start[j][1];
} else {
res[end[i][1]] = -1;
}
}
return res;
}
}