用两个二分板子,分别是找左边界和右边界的二分。
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型vector
* @param k int整型
* @return int整型
*/
int GetNumberOfK(vector<int>& nums, int k) {
// write code here
if(k > nums.back() || k < nums.front() || nums.size() == 0) return 0;
//用两个求左边界的二分
int l = 0;
int r = nums.size() - 1;
while(l < r){
int mid = l + r >> 1;
if(nums[mid] < k){
l = mid + 1;
}else{
r = mid;
}
}
int index1 = l;
l = 0, r = nums.size() - 1;
// 找出k的右边界
while(l < r){
int mid = l + r + 1 >> 1;
if(nums[mid] <= k){
l = mid;
}else{
r = mid - 1;
}
}
return l - index1 + 1;
}
};