我们先来看题目描述:
给你一个区间数组 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
- 每个间隔的起点都不相同
解决方案
方法一:二分查找
思路和算法
最简单的解决方案是对于集合中的每个区间,我们扫描所有区间找到其起点大于当前区间的终点的区间(具有最小差值),时间复杂度为 O(n2),在此我们不详细描述。
首先我们可以对区间 intervals 的起始位置进行排序,并将每个起始位置 intervals[i][0] 对应的索引 i 存储在数组 startIntervals 中,然后枚举每个区间 i 的右端点 ntervals[i][1],利用二分查找来找到大于等于 intervals[i][1] 的最小值 val 即可,此时区间 i 对应的右侧区间即为右端点 val 对应的索引。
代码:
Java
class Solution { public int[] findRightInterval(int[][] intervals) { int n = intervals.length; int[][] startIntervals = new int[n][2]; for (int i = 0; i < n; i++) { startIntervals[i][0] = intervals[i][0]; startIntervals[i][1] = i; } Arrays.sort(startIntervals, (o1, o2) -> o1[0] - o2[0]); int[] ans = new int[n]; for (int i = 0; i < n; i++) { int left = 0; int right = n - 1; int target = -1; while (left <= right) { int mid = (left + right) / 2; if (startIntervals[mid][0] >= intervals[i][1]) { target = startIntervals[mid][1]; right = mid - 1; } else { left = mid + 1; } } ans[i] = target; } return ans; } }