LeetCode 每日一题——436. 寻找右区间

1.题目描述

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] 有最小的“右”起点。

2.思路

2.1 代码

需要在给定的数组 intervals 中找到右区间，其实就是在 intervals 中找到第一位大于等于当前行第二位的行号，那么题目就可以简化为在二维数组 intervals 的第一列中找到某一个数。

这么一看这道题目就简单了，首先需要维护第一列的数和对应行的关系，这里用一个 Info 类进行维护，index 表示所在行，value 表示数值。然后使用数组存放每一列构造出来的 Info 对象，再根据 Info 的 value 将数组从小到大进行排序。

完成排序后，使用二分查找法在该数组中依次对每一行的第二位进行查找，最后将查找到的 Info 对象的 index 存入结果数组中。代码如下：

class Solution {
    public int[] findRightInterval(int[][] intervals) {
        int[] ans = new int[intervals.length];
        if (intervals.length == 1) {
            ans[0] = -1;
            return ans;
        }
        Info[] list = new Info[intervals.length];
        for (int i = 0; i < intervals.length; i++) {
            Info info = new Info(i, intervals[i][0]);
            list[i] = info;
        }
        Arrays.sort(list, Comparator.comparingInt(o -> o.value));
        for (int i = 0; i < intervals.length; i++) {
            ans[i] = getRightArea(list, intervals[i][1]);
        }
        return ans;
    }

    public int getRightArea(Info[] list, int num) {
        if (num > list[list.length - 1].value) {
            return -1;
        }
        int left = 0;
        int right = list.length - 1;
        while (left < right) {
            int mid = (left + right) / 2;
            if (list[mid].value > num) {
                right = mid;
            } else if (list[mid].value < num) {
                left = mid + 1;
            } else {
                return list[mid].index;
            }
        }
        return list[left].index;
    }
    private class Info {
        int value;
        int index;

        public Info(int index, int value) {
            this.index = index;
            this.value = value;
        }
    }
}

2.2 测试结果

通过测试

3.总结

• 使用 Info 类维护第一列和行的关系，并按照值进行排序
• 使用二分查找法对第二列的数进行查找