题目描述:
统计一个数字在排序数组中出现的次数。
思路:
看到排序数组,首先想到折半查找。折半查找到数字k第一次和最后一次出现的位置,两者相减即为出现的次数。时间复杂度O(logn)。若书序查找,时间复杂度则为O(n)。
代码:
语言C++,已通过牛课网在线测试。
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.size() == 0) return 0;
int first = getFirst(data, k);
int last = getLast(data, k);
return last - first + 1;
}
int getFirst(vector<int> data, int k){
int start = 0, end = data.size() - 1;
int mid = (start + end) >> 1;
while(start <= end){
if(data[mid] < k) start = mid + 1;
else end = mid - 1;//=k的话 end左移,找到第一个
mid = (start + end) >> 1;
}
return start;
}
int getLast(vector<int> data, int k){
int start = 0, end = data.size() - 1;
int mid = (start + end) >> 1;
while(start <= end){
if(data[mid] <= k) start = mid + 1;//=k的话 start右移 找到最后一个
else end = mid - 1;
mid = (start + end) >> 1;
}
return end;
}
};