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

题目

统计一个数字在排序数组中出现的次数。

思路

 使用二分查找分别找到左边界left和右边界right,所以数字target的数量就是right - left - 1

代码

class Solution {
    
    public int search(int[] nums, int target) {
    
        // 搜索右边界 right
        int i = 0,j = nums.length - 1;
        while(i <= j)
        {
    
            int m = (i + j) / 2;
            if(nums[m] < target) 
            {
    
                i = m + 1;// 说明target在闭区间[m + 1,j] 上
            }
            else if(nums[m] ==  target)
            {
    
                i = m + 1;  // 因为查找右边界 所以 右边界一定在闭区间[m + 1, j]中
            }
            else
            {
    
                j = m  -1;// target在闭区间[i,m - 1]中 
            }
        }
        int right = i; // 定位右边界 此时right指向的数字 不是target
        if(j >= 0 && nums[j] != target) return 0;

        // 搜索左边界 right
        i = 0;
        j = nums.length - 1;

        while(i <= j)
        {
    
            int m = (i + j) / 2;
            if(nums[m]  < target)
            {
    
                i = m + 1;
            }
            else{
    
                j = m - 1;
            }
        }
        int  left = j;// 左边界指向的数字不是target
        return right - left - 1;
    }
}