题目来源:力扣
题目
给你一个区间数组 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] 有最小的“右”起点。
提示:
1 <= intervals.length <= 2 * 104
intervals[i].length == 2
-106 <= starti <= endi <= 106
每个间隔的起点都 不相同
题解
题解一(两层遍历)
public static int[] findRightInterval(int[][] intervals) {
int[] result = new int[intervals.length];
for(int i = 0;i < intervals.length;i++) {
int start = intervals[i][0];
int end = intervals[i][1];
if(start == end) {
result[i] = i;
continue;
}else {
int max = Integer.MAX_VALUE;
int index = -1;
for(int j = 0;j < intervals.length;j++) {
if(intervals[j][0] >= end && intervals[j][0] < max) {
max = intervals[j][0];
index = j;
}
}
result[i] = index;
}
}
return result;
}
题解二(先快排,再二分查找)
public static int[] findRightInterval(int[][] intervals) {
int n = intervals.length;
int[][] startInterval = new int[n][2];
for(int i = 0;i < n;i++){
startInterval[i][0] = intervals[i][0];
startInterval[i][1] = i;
}
//快排
quickSort(startInterval,0,startInterval.length - 1);
int[] re = new int[n];
//二分查找
for(int i = 0;i < intervals.length;i++){
int result = -1;
int left = 0;
int right = intervals.length - 1;
while(left <= right){
int mid = (left + right) / 2;
if(startInterval[mid][0] >= intervals[i][1]){
right = mid - 1;
result = startInterval[mid][1];
}else{
left = mid + 1;
}
}
re[i] = result;
}
return re;
}
public static void quickSort(int[][] arr,int start,int end){
if(start < end){
int pos = qSort(arr,start,end);
quickSort(arr,start,pos - 1);
quickSort(arr,pos + 1,end);
}
}
public static int qSort(int[][] arr,int start,int end){
while(start < end){
while(arr[start][0] <= arr[end][0] && start < end)end--;
int[] temp = arr[start];
arr[start] = arr[end];
arr[end] = temp;
while(arr[start][0] <= arr[end][0] && start < end)start++;
temp = arr[start];
arr[start] = arr[end];
arr[end] = temp;
}
return start;
}
题解三(先快排,再用双指针查找)
public static int[] findRightInterval(int[][] intervals) {
int n = intervals.length;
int[][] startInterval = new int[n][2];
int[][] endInterval = new int[n][2];
for(int i = 0;i < intervals.length;i++){
startInterval[i][0] = intervals[i][0];
endInterval[i][0] = intervals[i][1];
startInterval[i][1] = i;
endInterval[i][1] = i;
}
quickSort(startInterval,0,intervals.length - 1);
quickSort(endInterval,0,intervals.length - 1);
int[] re = new int[n];
int i = 0;
for(int[] end : endInterval){
while(i < n && startInterval[i][0] < end[0])i++;
if(i == n){
re[end[1]] = -1;
}else {
re[end[1]] = startInterval[i][1];
}
}
return re;
}
public static void quickSort(int[][] arr,int start,int end){
if(start < end){
int pos = qSort(arr,start,end);
quickSort(arr,start,pos - 1);
quickSort(arr,pos + 1,end);
}
}
public static int qSort(int[][] arr,int start,int end){
while(start < end){
while(arr[start][0] <= arr[end][0] && start < end)end--;
int[] temp = arr[start];
arr[start] = arr[end];
arr[end] = temp;
while(arr[start][0] <= arr[end][0] && start < end)start++;
temp = arr[start];
arr[start] = arr[end];
arr[end] = temp;
}
return start;
}