1,题目要求
统计一个数字在排序数组中出现的次数。
2,题目思路
对于这道题,要求找出一个经过排序的数组中某个数字出现的次数。
如果数组是无序的,我们可以直接遍历、统计即可。
而在这道题中,给定的数组是有序的,因此,我们可以使用二分搜索的思想,在该数组中找到第一个和最后一个等于k的数字即可。
需要注意的是,二分搜索的条件判断:
- 在搜索第一个等于k的数字的位置时,利用data[mid] < k的判断方法,这样,最后start会落在第一个位置位置上
- 在搜索最后一个等于k的数字位置时,利用data[mid] <=k的判断方法,这样砸搜索的过程中会不断向后探索,最后当start和end都相等时,mid是落在最后一个等于k的后一个位置上,因此,在条件判断上,end = mid-1,自然就落在对应的最后一个等于k的位置上了。
3,代码实现
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int lower = firstK(data, k);
int upper = lastK(data, k);
return upper - lower + 1;
}
private:
int firstK(vector<int> &data, int k){
int start = 0, end = data.size()-1;
while(start <= end){
int mid = start+(end - start)/2;
if(data[mid]<k)
start = mid + 1;
else
end = mid - 1;
}
return start;
}
int lastK(vector<int> &data, int k){
int start = 0, end = data.size()-1;
while(start <= end){
int mid = start+(end - start)/2;
if(data[mid]<=k)
start = mid + 1;
else
end = mid - 1;
}
return end;
}
};