技術標籤：劍指offer二分法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 <= 陣列長度 <= 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;
    }