Java实现 LeetCode 436 寻找右区间

436. 寻找右区间

给定一组区间，对于每一个区间 i，检查是否存在一个区间 j，它的起始点大于或等于区间 i 的终点，这可以称为 j 在 i 的“右侧”。

对于任何区间，你需要存储的满足条件的区间 j 的最小索引，这意味着区间 j 有最小的起始点可以使其成为“右侧”区间。如果区间 j 不存在，则将区间 i 存储为 -1。最后，你需要输出一个值为存储的区间值的数组。

注意:

你可以假设区间的终点总是大于它的起始点。

你可以假定这些区间都不具有相同的起始点。

示例 1:

输入: [ [1,2] ]

输出: [-1]

解释:集合中只有一个区间，所以输出-1。

示例 2:

输入: [ [3,4], [2,3], [1,2] ]

输出: [-1, 0, 1]

解释:对于[3,4]，没有满足条件的“右侧”区间。

对于[2,3]，区间[3,4]具有最小的“右”起点;

对于[1,2]，区间[2,3]具有最小的“右”起点。

示例 3:

输入: [ [1,4], [2,3], [3,4] ]

输出: [-1, 2, -1]

解释:对于区间[1,4]和[3,4]，没有满足条件的“右侧”区间。

对于[2,3]，区间[3,4]有最小的“右”起点。

class Solution {
    public int[] findRightInterval(int[][] intervals) {
        int length = intervals.length;
        int[] res = new int[length];
        MyNode[] list = new MyNode[length];
        for (int i = 0; i < length; i++) {
            list[i] = new MyNode(intervals[i][0], intervals[i][1], i);
        }
        Arrays.sort(list, new Comparator<MyNode>() {
            public int compare(MyNode a, MyNode b) {
                return a.start - b.start;
            }
        });
        for (int i = 0; i < length; i++) {
            res[list[i].index] = greater(list, 0, length - 1, list[i].end);
        }
        return res;
    }

    class MyNode {
        int start;   // 区间的起点
        int end;     // 区间的终点
        int index;   // 在原数组的下标

        public MyNode(int start, int end, int index) {
            this.start = start;
            this.end = end;
            this.index = index;
        }
    }

    // 二分查找第一个区间的起点大于等于target的数组元素的下标，数组是升序排序，范围是[left,right]
    private int greater(MyNode[] list, int left, int right, int target) {
        while (left <= right) {
            int mid = (left + right) >>> 1;
            if (list[mid].start < target) {
                left = mid + 1;
            } else {
                if (mid - 1 >= left && list[mid - 1].start >= target) {
                    right = mid - 1;
                } else {
                    return list[mid].index;
                }
            }
        }
        return -1;        // list[right].start<target
    }
}