题目描述
统计一个数字在排序数组中出现的次数。
分析:这个题目并没有说是升序的数组,但所有的解法都是默认升序的,这就有点坑了,默认升序的话会少很多麻烦。然后这道题有挺多思想的,最暴力的就是直接遍历,复杂度O(n),然后就是利用STL中的count函数,或者lower_bound和upper_bound函数,再然后是一个比较有趣的思想,因为是整数数组,可以用二分查找的方法找k+0.5与k-0.5在数组中的位置,就会消除重复数字的影响了。然后就是用的最多的二分查找,不过因为有重复数字的影响,需要找寻两次,一次寻找第一个出现的位置,一次寻找最后一次出现的位置。
代码:二分查找寻找上下界,默认数组升序
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.empty())
return 0;
return lastK(data,k)-firstK(data,k)+1;
}
int firstK(vector<int> data,int k){
int start=0,end=data.size()-1;
int mid=(end+start)/2;
while(start<=end){
if(data[mid]<k)
start=mid+1;
else
end=mid-1;
mid=(end+start)/2;
}
return start;
}
int lastK(vector<int> data,int k){
int start=0,end=data.size()-1;
int mid=(end+start)/2;
while(start<=end){
if(data[mid]<=k)
start=mid+1;
else
end=mid-1;
mid=(end+start)/2;
}
return end;
}
};