开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第27天,点击查看活动详情
给你一个区间数组 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
- 每个间隔的起点都 不相同
思路
本题本质就是在数组中查找大于等于endi的最小值,涉及值查找的,基本上都可以用二分查找来求解。由于本题求的是右侧区间的下标,我们先把小标和区间绑定到以前,然后区间按照start值升序排序。遍历区间,在当前区间之后而查找大于等于endi的最小start值,并返回其下标,如果没找到,返回-1,最后把返回值用数组存储并按照区间下标排序,就是我们要求的解了。
解题
/**
 * @param {number[][]} intervals
 * @return {number[]}
 */
var findRightInterval = function (intervals) {
  const len = intervals.length;
  const res = Array.from({ length: len });
  intervals.forEach((item, i) => (item[2] = i));
  intervals.sort((a, b) => a[0] - b[0]);
  const maxStart = intervals[len - 1][0]
  for (let i = 0; i < len; i++) {
    const end = intervals[i][1];
    if (maxStart < end) {
      res[intervals[i][2]] = -1;
    } else {
      let left = i;
      let right = len - 1;
      while (left <= right) {
        const mid = (left + right) >> 1;
        const start = intervals[mid][0];
        if (start >= end) {
          right = mid - 1;
        } else if (start < end) {
          left = mid + 1;
        }
      }
      res[intervals[i][2]] = intervals[right + 1][2];
    }
  }
  return res;
};