leetcode 34 二分搜索变种

思路：

题意很明显了，要使用二分搜索在O(lgn)时间内找到第一次出现和最后一次出现的目标下标，如果没有出现，则返回-1。

一直以来都只会写最基本的二分查找（找到就停那种），要么就是为了省事直接用Arrays.binarySearch(arr, target);

这次自己写了下，感觉也不是很难。

 

在最基本的版本上增加两个东西：

1. 不是找到就停，而是继续压缩查找空间。first就压缩上界；last就压缩下界。

2. 由于不知道何时停止，需要一个变量ind来不断更新target对应的下标。

 

class Solution {
    public int[] searchRange(int[] nums, int target) {
        // return new int[]{firstBinarySearch(nums, target),
        //                 lastBinarySearch(nums, target)};
        return new int[]{specialBinarySearch(nums, target, true),
                        specialBinarySearch(nums, target, false)};
    }
    
    public int firstBinarySearch(int[] nums, int target) {
        int low = 0, high = nums.length - 1;
        int ind = -1;
        while (low <= high) {
            int mid = (low + high) / 2;
            if (nums[mid] < target)
                low = mid + 1;
            else{
                ind = nums[mid] == target ? mid : ind;
                high = mid - 1;
            }   
        }
        return ind;
    }
    
    public int lastBinarySearch(int[] nums, int target) {
        int low = 0, high = nums.length - 1;
        int ind = -1;
        while (low <= high) {
            int mid = (low + high) / 2;
            if (nums[mid] > target)
                high = mid - 1;
            else{
                ind = nums[mid] == target ? mid : ind;
                low = mid + 1;
            }   
        }
        return ind;
    }
    
    public int specialBinarySearch(int[] nums, int target, boolean isFirst) {
        int low = 0, high = nums.length - 1;
        int ind = -1;
        while (low <= high) {
            int mid = (low + high) / 2;
            if (nums[mid] < target)
                low = mid + 1;
            else if (nums[mid] > target)
                high = mid - 1;
            else {
                ind = mid;
                high = isFirst ? mid - 1 : high;
                low = !isFirst ? mid + 1 : low;
            }
        }
        return ind;
    }
}

 

这里还写了一个最终版本specialBinarySearch，将两种情况合并到一起判断。是不是和原来的基础版很接近？