题目描述

blablabla

C++ 代码

class Solution {
public:
    int getNumberOfK(vector<int>& nums , int k) {
        if (!nums.size()) return 0;
        if (k < nums[0] || k > nums[nums.size() - 1]) return 0;
        int n = nums.size();
        int first = getFirst(nums, 0, k, n - 1);
        int last = getLast(nums, 0, k, n - 1);
        return last - first + 1;
    }
    int getFirst(vector<int>& nums, int l, int k, int r) {
        while(l < r) {
            int mid = l + r >> 1;//大于等于k的最小值。
            if (nums[mid] >= k) r = mid;
            else l = mid + 1;
        }
        return l;
    }
    int getLast(vector<int>& nums, int l, int k, int r) {
        while(l < r) {
            int mid = l + r + 1 >> 1;//小于等于k的最大值。所以找不到k的时候，last比first小1，返回的结果也是正确的。
            if (nums[mid] <= k) l = mid;
            else r = mid - 1;
        }
        return l;
    }
};