leetcode 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 len = intervals.length;
        if(len == 0){
            return new int[0];
        }
        Map<Integer, Integer> hashMap = new HashMap<>(len);

        int[] arr = new int[len];
        int[] res = new int[len];
        for(int i = 0; i < len; i++){
            hashMap.put(intervals[i][0], i);
            arr[i] = intervals[i][0];
        }

        Arrays.sort(arr);

        for(int i = 0; i < len; i++){
            int index = binarySearch(arr, intervals[i][1]);
            if(index == -1){
                res[i] = -1;
            }
            else{
                res[i] = hashMap.get(arr[index]);
            }
        }
        return res;
    }
    //查找第一個大於等於target的元素索引
    private int binarySearch(int[] arr, int target){
        int len = arr.length;
        if(arr[len - 1] < target){
            return -1;
        }
        int L = 0, R = len - 1;
        while(L < R){
            int mid = (L + R) >>> 1;
            if(target > arr[mid]){
                L = mid + 1;
            }
            else{
                R = mid;
            }
        }
        return L;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章