[Leetcode] 436. Find Right Interval

leetcode題目
題意:給定一組區間,求給定區間 在這些區間中 在給定區間右側,且與給定區間距離最近區間的位置。

思路:
把給定區間的右邊界與這些區間中的左邊界進行比較,選擇出,左邊界比給定區間右邊界大,且距給定區間最近區間。獲取該區間的位置。

具體代碼實現:

利用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; }
 * }
 */
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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章