ACWING67. 数字在排序数组中出现的次数(剑指offer,二分)

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

例如输入排序数组[1, 2, 3, 3, 3, 3, 4, 5]和数字3,由于3在这个数组中出现了4次,因此输出4。

样例
输入:[1, 2, 3, 3, 3, 3, 4, 5] , 3

输出:4

class Solution {
public:
    int LowerBound(vector<int>& nums,int k) {
        int l = 0,r = nums.size() - 1;
        int ans = 0;
        while(l <= r) {
            int mid = (l + r) >> 1;
            if(nums[mid] >= k) {
                r = mid - 1;
                ans = mid;
            }
            else l = mid + 1;
        }
    }
    
    int UpperBound(vector<int>& nums,int k) {
        int l = 0,r = nums.size() - 1;
        int ans = 0;
        while(l <= r) {
            int mid = (l + r) >> 1;
            if(nums[mid] <= k) {
                l = mid + 1;
                ans = mid;
            }
            else r = mid - 1;
        }
    }
    
    int getNumberOfK(vector<int>& nums , int k) {
        if(!nums.size()) return 0;
        int k1 = LowerBound(nums,k);
        int k2 = UpperBound(nums,k);
        return k2 - k1;
    }
};
发布了843 篇原创文章 · 获赞 28 · 访问量 4万+

猜你喜欢

转载自blog.csdn.net/tomjobs/article/details/104979115