436. 寻找右区间

题目

给你一个区间数组 intervals ，其中 intervals[i] = [starti, endi] ，且每个 starti 都 不同 。

区间 i 的 右侧区间 可以记作区间 j ，并满足 startj >= endi ，且 startj 最小化 。

返回一个由每个区间 i 的 右侧区间 的最小起始位置组成的数组。如果某个区间 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 * 10^4
intervals[i].length == 2
-10^6 <= starti <= endi <= 10^6
每个间隔的起点都 不相同

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-right-interval
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

将数组按start值排序，对于每个end，查找右侧第一个不小于自身的start值，并获取其下标。

代码

class Solution {
    public int[] findRightInterval(int[][] intervals) {
        int m = intervals.length;
        // 因为最后要返回下标，所以不能对原数组进行排序
        int[][] arr = new int[m][2];
        for (int i = 0; i < m; i++) {
            // 0下标为原始下标
            arr[i][0] = i;
            // 1下标为start值
            arr[i][1] = intervals[i][0];
        }
        // 按start进行排序
        Arrays.sort(arr, (a, b) -> a[1] - b[1]);
        int[] ans = new int[m];
        for (int i = 0; i < m; i++) {
            // 对于每个end，查找右侧第一个不小于自身的start
            ans[i] = bsearch(arr, intervals[i][1]);
        }
        return ans;
    }
    // 在按arr[1]升序的arr数组中查找第一个不小于num的arr[1]，并返回其下标arr[0]
    private int bsearch(int[][] arr, int num) {
        int l = 0;
        int r = arr.length - 1;
        while (l < r) {
            int mid = l + (r - l) / 2;
            if (arr[mid][1] < num) {
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        // 循环结束，需要后判，可能不存在
        if (arr[l][1] >= num) {
            return arr[l][0];
        }
        return -1;
    }
}