日新刷题 - 436. 寻找右区间

103 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第17天,点击查看活动详情

一 描述

436. 寻找右区间 - 力扣(LeetCode)

给你一个区间数组 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 * 10^4
  • intervals[i].length == 2
  • -10^6 <= starti <= endi <= 10^6
  • 每个间隔的起点都 不相同

二 分析

如果有区间,intervals[i] = [starti, endi],那么搜索其他区间j,并满足 startj >= endi ,且 startj 最小化

这里很容易想到二分搜索,但是数组是无序的,我们需要将数组排序,但是排序后我们将打乱原数组的元素顺序。

题目的要求是我们返回对应区间的结果,那么这里我们可以把原数组扩容至长度4,copy[2]记录该元素在原数组的位置,copy[3]记录最终结果。

三 答案

class Solution {
    public int[] findRightInterval(int[][] intervals) {
        int[][] copy = new int[intervals.length][4];
        for (int i = 0; i < intervals.length; i++) {
            copy[i][0] = intervals[i][0];
            copy[i][1] = intervals[i][1];
            copy[i][2] = i;
        }
        Arrays.sort(copy, Comparator.comparingInt(a -> a[0]));
        for (int i = 0; i < copy.length; i++) {
            copy[i][3] = find(copy, copy[i][1]);
        }
        int[] ans = new int[intervals.length];
        for (int i = 0; i < ans.length; i++) {
            ans[copy[i][2]] = copy[i][3];
        }
        return ans;
    }

    //按照从小到大的顺序寻找遍历元素ints[i][0],找到第一个大于或者等于target的元素
    private int find(int[][] ints, int target) {
        int l = 0, r = ints.length - 1;
        while (l < r) {
            int mid = (r - l) / 2 + l;
            if (ints[mid][0] < target) {
                //左区间都不符合题意
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        //注意这里要返回该元素在原数组的位置,而不是返回指针所在的当前位置
        return ints[l][0] >= target ? ints[l][2] : -1;
    }
}