题目描述
统计一个数字在排序数组中出现的次数。
解题思路:
因为数组有序,可用二分查找法,找到第一个k之后,用双指针向两边扩展
代码实现:
class Solution {
/*
因为数组有序,可用二分查找法,找到第一个k之后,用双指针向两边扩展
*/
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.size() == 1 && data[0] == k){
return 1;
}
int count = 0;
int low = 0,high = data.size()-1;
while(low < high){
int mid = (low+high) / 2;
if(data[mid] == k){
//找到第一个k之后,用双指针向两边扩展
count++;
int up = mid+1;
int down = mid-1;
while(data[up]==k){
count++;
up++;
}
while(data[down]==k){
count++;
down--;
}
return count;
}else if(data[mid] > k){
high = mid - 1;
}else{
low = mid + 1;
}
}
return 0;
}
};