数字在排序数组中出现的次数

class Solution {
public:
//二分找出k第一次,最后一个k后面的下标,相减就是答案
    int getNumberOfK(vector<int>& nums , int k) {
        int n=nums.size();
        if(!n)  return 0;
        //找到第一个≥k的下标
        int l=0,r=n-1;
        while(l<r)
        {
            int mid=l+r>>1;
            if(nums[mid]<k) l=mid+1;
            else r=mid;
        }
        int start=l;
        if(nums[start]!=k)  return 0;
        //找到最后一个≤k的下标
        l=0,r=n-1;
        while(l<r)
        {
            int mid=l+r+1>>1;
            if(nums[mid]>k) r=mid-1;
            else l=mid;
        }
        int end=l;
        return end-start+1;
    }
};
posted @ 2023-04-28 15:19  穿过雾的阴霾  阅读(17)  评论(0)    收藏  举报