题目
统计一个数字在排序数组中出现的次数。
例如输入排序数组[1, 2, 3, 3, 3, 3, 4, 5]和数字3,由于3在这个数组中出现了4次,因此输出4。
样例
输入:[1, 2, 3, 3, 3, 3, 4, 5] , 3
输出:4
解法:二分
数组已经排好序,那么直接分别查找上下界即可
时间复杂度O(logn),空间复杂度O(1)
class Solution {
public:
int getNumberOfK(vector<int>& nums , int k) {
if (nums.empty()) return 0;
if (k < nums[0] || k > nums[nums.size() - 1]) return 0;
int l = 0, r = nums.size() - 1;
while (l < r) {
int mid = l + r + 1 >> 1;
if (nums[mid] <= k) l = mid;
else r = mid - 1;
}
int right = l;
l = 0, r = nums.size() - 1;
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] >= k) r = mid;
else l = mid + 1;
}
int left = l;
return right - left + 1;
}
};