Leetcode 436.寻找右区间

寻找右区间

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

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

注意:

1. 你可以假设区间的终点总是大于它的起始点。
2. 你可以假定这些区间都不具有相同的起始点。

示例 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]有最小的"右"起点。

解题思路

利用java TreeMap的性质，把所有区间的左边界作为key值，所在位置作为value值，保存在map中，利用TreeMap中已有的ceilingKey(key k)方法，直接获取与给定key大且最近的key值，然后通过key值得到位置。

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
import java.util.TreeMap;

public class Solution{
    public int[] findRightInterval(Interval[] intervals){
        int len=intervals.length;
        int nums[]=new int[len];
        TreeMap<Integer,Integer> map=new TreeMap<Integer,Integer>();
        for(int i=0;i<len;i++){
            map.put(intervals[i].start,i);
        }
        for(int i=0;i<len;i++){
            Integer num=map.ceilingKey(intervals[i].end);
            if(num==null){
                nums[i]=-1;
            }else{
                nums[i]=map.get(num);
            }
        }
        return nums;
    }
}