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

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

题目描述

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

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

提示：

0 <= nums.length <= 10⁵
-10⁹ <= nums[i] <= 10⁹
nums 是一个非递减数组
-10⁹ <= target <= 10⁹

源码

思路：

• 二分查找，寻找是否存在target数字
• 在存在的基础上，使用扩展边界

class Solution {
    public int search(int[] nums, int target) {
        int n = nums.length;
        int left = 0;
        int right = n - 1;
        int count = 0;
        while(left <= right){
            int mid = left + (right - left) / 2;
            if(nums[mid] < target){
                left = mid + 1;
            }else if(nums[mid] > target){
                right = mid - 1;
            }else{
                if(nums[mid] == target){
                    int i = mid;
                    int j = mid;
                    while(i > 0 && nums[i - 1] == target){
                        i --;
                    }
                    while(j < n - 1 && nums[j + 1] == target){
                        j ++;
                    }
                    return j - i + 1;
                }
            }
        }
        return 0;
    }
}