问题描述
统计一个数字在排序数组中出现的次数。
class Solution {
public:
int GetNumberOfK(vector data ,int k) {}
};分析
最笨最耗时的方法,从头遍历,如果等于k,开始计数,不等于k时返回计数结果。但是这样时间复杂度为o(n),比较麻烦。
比较聪明的方法1:因为排序数组,所以二分法找到第一个k,记下下标i,然后同样方法找到最后一个k,记下下标j,最后返回j-i+1。但是这样写了两个辅助方法后,又比较麻烦,提交后超时了。这个时候我们其实可以想到完全没必要找两个k的位置,一个头或者一个尾就足够了。所以有了比较聪明的方法2:
如果找的是头那个k的下标,就向后查找,等于k计数器加1,不等就返回计数器的值。找的是尾的那个k的下标时同理。
当然这个方法2也可以用二分法直接找k,随便找到哪一个,然后向左右遍历,找到最左最右端的k下标,然后再减。
所以就有很多种代码可以达到这个目的,下面只列最后一种代码。代码
class Solution {
public:
int GetK(vector<int>data,int k){
int size=data.size();
if(data.empty() || size==0){
return-1;
}
int first=0;
int end=size-1;
int middle;
while(first<=end){
middle=(first+end)/2;
if(data[middle]<k){
first=middle+1;
}else if(data[middle] >k){
end=middle-1;
}else{
return middle;
}
}
return -1;
}
int GetNumberOfK(vector<int> data ,int k) {
int position=GetK(data,k);
if(position==-1){
return 0;
}
int i=position;
int j=position;
int size=data.size();
while(i>=0 && data[i]==k)
i--;
while(j<size&&data[j]==k)
j++;
return j-i-1;
}
};