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;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章