描述
给定一个长度为 n 的非降序数组和一个非负数整数 k ,要求统计 k 在数组中出现的次数
数据范围:0 \le n \le 1000 , 0 \le k \le 1000≤n≤1000,0≤k≤100,数组中每个元素的值满足 0 \le val \le 1000≤val≤100
要求:空间复杂度 O(1),时间复杂度 O(logn)
示例1
输入:
[1,2,3,3,3,3,4,5],3
复制返回值:
4
复制
示例2
输入:
[1,3,4,5],6
复制返回值:
0
思路:寻找上界和下界,差值就位次数
下界定义为:如果存在目标值,则指向第一个目标值,否则,如果不存在, 则指向大于目标值的第一个值。
上界定义为:不管目标值存在与否,都指向大于目标值的第一个值。
如下图所示:
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.size()==0||k>data[data.size()-1] || k<data[0]) return 0;
int l=0,r=0;
int left=0,right=data.size();
while(left<right){ //寻找下界
int mid=(left+right)/2;
if(data[mid]<k){
left=mid+1;
}else{
right=mid;
}
}
l=left;
left=0,right=data.size();
while(left<right){ //寻找上界
int mid=(left+right)/2;
if(data[mid]<=k){
left=mid+1;
}else{
right=mid;
}
}
r=left;
return r-l;
}
};