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

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

统计一个数字在排序数组中出现的次数
示例：输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

思路：
1.为了利用数租已经排序好的条件，考虑 target 在数组中出现的次数，其实我们要找的就是数组中第一个大于等于target的下标，记为 leftindex和第一个大于target的下标，记为rightIndex。所以target出现的次数为rightIndex-leftindex-1；
为了代码的复用，我们定义 binarySearch(nums, target, lower) 表示在 nums 数组中二分查找 target 的位置，如果lower 为true，则查找第一个大于等于 target 的下标，否则查找第一个大于 target 的下标。

class Solution {
public:
    int binarySearch(vector<int>& nums, int target, bool lower) {
        int left = 0;
        int right =nums.size() - 1;
        int n  = nums.size();
        while (left <= right) {
            int mid = (left + right) / 2;
            if (nums[mid] > target || (lower && nums[mid] >= target)) {
                right = mid - 1;
                n = mid;
            } else {
                left = mid + 1;
            }
        }
        return n;
    }

    int search(vector<int>& nums, int target) {
        int leftIdx = binarySearch(nums, target, true);
        int rightIdx = binarySearch(nums, target, false) - 1;
        if (leftIdx <= rightIdx && rightIdx < nums.size() && nums[leftIdx] == target && nums[rightIdx] == target) {
            return rightIdx - leftIdx + 1;
        }
        return 0;
    }
};