剑指 Offer 53 - I. 在排序数组中查找数字 I（Java、二分查找）

原题地址

剑指 Offer 53 - I. 在排序数组中查找数字 I

题目描述

题目思路

• 在排序数组中进行搜索时，经常使用到二分查找。对于本题，可以通过两次二分查找来分别确定左右边界，最后返回右边界 - 左边界 + 1即为目标数字出现次数。

二分查找

• 在升序数组中查找目标值索引时，可以使用二分查找。二分查找是指在查找范围为[left,right]的数组中，每次选择该区间的中点（mid)，判断需要查找的值与nums[mid]的大小。若大于nums[mid]，则位于mid右边，反之则位于左边。
• 由于每次查找都会将查找范围缩小一半，因此二分查找的时间复杂度是 O(log⁡n)，其中n是数组的长度。
• 在确定中点（mid）时，（left + right）/ 2 可以会造成溢出，因此可以考虑改为（left + right） >> 1。

代码实现

二分查找

class Solution {
    int mid ;
    public int search(int[] nums, int target) {
        int i = 0;
        int j = nums.length - 1;
        //确定右边界
        while(i <= j) {
            mid = (i + j) >> 1;
            if(nums[mid] <= target) {
                i = mid + 1;
            }else {
                j = mid - 1;
            }
        }

        int right = j;

        i = 0;
        j = nums.length - 1;
        //确定左边界
        while(i <= j) {
            mid = (i + j) >> 1;
            if(nums[mid] >= target) {
                j = mid - 1;
            }else {
                i = mid + 1;
            }
        }

        int left = i;
        return right - left + 1;

    }
}

二分查找优化

class Solution {
    int mid ;
    public int search(int[] nums, int target) {
        int i = 0;
        int j = nums.length - 1;
        //确定右边界
        while(i <= j) {
            mid = (i + j) >> 1;
            if(nums[mid] <= target) {
                i = mid + 1;
            }else {
                j = mid - 1;
            }
        }

        int right = j;

		/*查找完右边界后，可判断数组中是否包含 targettargettarget，
		若不包含则直接提前返回 0 ，无需后续查找左边界。*/
        if(j >= 0 && nums[j] != target) return 0;
        
        i = 0;
        j = nums.length - 1;
        //确定左边界
        while(i <= j) {
            mid = (i + j) >> 1;
            if(nums[mid] >= target) {
                j = mid - 1;
            }else {
                i = mid + 1;
            }
        }

        int left = i;
        return right - left + 1;

    }
}

二分查找简化

class Solution {
	/*只查询右边界的二分查找
	只需查询目标值的右边界与目标值-1的右边界，即可确定目标值个数*/
    public int search(int[] nums, int target) {
        return helper(nums, target) - helper(nums, target - 1);
    }
    int helper(int[] nums, int tar) {
        int i = 0, j = nums.length - 1;
        while(i <= j) {
            int m = (i + j) / 2;
            if(nums[m] <= tar) i = m + 1;
            else j = m - 1;
        }
        return i;
    }
}