LeetCode~436. 寻找右区间

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

题目描述

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

区间 i 的 右侧区间 可以记作区间 j ，并满足 start$_j$ >= end$_i$ ，且 start$_j$ 最小化 。

返回一个由每个区间 i 的 右侧区间 在 intervals 中对应下标组成的数组。如果某个区间 i 不存在对应的 右侧区间 ，则下标 i 处的值设为 -1 。

示例

输入： intervals = [[1,2]]
输出： [-1]
解释： 集合中只有一个区间，所以输出-1。

输入：intervals = [[3,4],[2,3],[1,2]]
输出：[-1,0,1]
解释：对于 [3,4] ，没有满足条件的“右侧”区间。
对于 [2,3] ，区间[3,4]具有最小的“右”起点;
对于 [1,2] ，区间[2,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$ <= start$_i$ <= end$_i$ <= 10$^6$
• 每个间隔的起点都 不相同

哈希树

class Solution {
    public int[] findRightInterval(int[][] intervals) {
        TreeMap<Integer, Integer> treeMap = new TreeMap<>();
        int n = intervals.length;
        for(int i = 0; i < n; ++i){
            treeMap.put(intervals[i][0], i);
        }

        int[] res = new int[n];

        for(int i = 0; i < n; ++i){
            Map.Entry<Integer, Integer> entry = treeMap.ceilingEntry(intervals[i][1]);
            if(null != entry){
                res[i] = entry.getValue();
            }else{
                res[i]= -1;
            }
        }

        return res;
    }
}