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

剑指 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 <= 数组长度 <= 50000

基本思路：二分法，时间复杂度O(nlgn)

    int binarySearch(vector<int> &nums,int l,int r,int &target,bool flag){
        if(l>r)
            return -1;
        int ans=r+1;  //注意ans的初始化，target可能是最后一个元素，此时ans==r+1;
        while(l<=r){
            int mid=(r-l)/2+l;
            if(nums[mid]>target||(flag&&nums[mid]==target)){  //nums[i]>=target ,获得左边界，nums[i]>target,获得有边界+1
                ans=mid;
                r=mid-1;
            }
            else{
                l=mid+1;
            }
        }

        return ans;
        
    }
    int search(vector<int>& nums, int target) {
        int len=nums.size();
        if(len<1||target>nums[len-1]||target<nums[0])
            return 0;
        int leftIdx=binarySearch(nums,0,nums.size()-1,target,true);
        int rightIdx=binarySearch(nums,leftIdx,nums.size()-1,target,false);  
        //cout<<leftIdx<<"  "<<rightIdx<<endl;
        if(leftIdx>=0&&leftIdx<rightIdx&&rightIdx<=len){  //rightIdx，实际上指向的是最后的target的下一个，所以rightIdx==len
            return rightIdx-leftIdx;
        }

        return 0;
    }