題目描述
統計一個數字在排序數組中出現的次數。
分析:這個題目並沒有說是升序的數組,但所有的解法都是默認升序的,這就有點坑了,默認升序的話會少很多麻煩。然後這道題有挺多思想的,最暴力的就是直接遍歷,複雜度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;
}
};