[Leetcode] Search for a Range 尋找區(qū)間

zsirfs 發(fā)布于2019-08-14 12:42 / 919人閱讀

摘要：二分搜索復(fù)雜度時(shí)間空間思路其實(shí)就是執(zhí)行兩次二分搜索，一次專門找左邊邊界，一次找右邊邊界。如果找右邊邊界，則要判斷右邊一位的數(shù)是否相同。

Search for a Range

Given a sorted array of integers, find the starting and ending position of a given target value.
Your algorithm"s runtime complexity must be in the order of O(log n).
If the target is not found in the array, return [-1, -1].
For example, Given [5, 7, 7, 8, 8, 10] and target value 8, return [3, 4].

二分搜索 復(fù)雜度

時(shí)間 O(logN) 空間 O(1)

思路

其實(shí)就是執(zhí)行兩次二分搜索，一次專門找左邊邊界，一次找右邊邊界。特別的，如果找左邊邊界時(shí)，要看mid左邊一位的的數(shù)是否和當(dāng)前mid相同，如果相同要繼續(xù)在左半邊二分搜索。如果找右邊邊界，則要判斷右邊一位的數(shù)是否相同。

代碼

public class Solution {
    public int[] searchRange(int[] nums, int target) {
        // 找到左邊界
        int front = search(nums, target, "front");
        // 找到右邊界
        int rear = search(nums, target, "rear");
        int[] res = {front, rear};
        return res;
    }
    
    public int search(int[] nums, int target, String type){
        int min = 0, max = nums.length - 1;
        while(min <= max){
            int mid = min + (max - min) / 2;
            if(nums[mid] > target){
                max = mid - 1;
            } else if(nums[mid] < target){
                min = mid + 1;
            } else {
                // 對(duì)于找左邊的情況，要判斷左邊的數(shù)是否重復(fù)
                if(type == "front"){
                    if(mid == 0) return 0;
                    if(nums[mid] != nums[mid - 1]) return mid;
                    max = mid - 1;
                } else {
                // 對(duì)于找右邊的情況，要判斷右邊的數(shù)是否重復(fù)
                    if(mid == nums.length - 1) return nums.length - 1;
                    if(nums[mid] != nums[mid + 1]) return mid;
                    min = mid + 1;
                }
            }
        }
        //沒找到該數(shù)返回-1
        return -1;
    }
}