题目描述
统计一个数字在排序数组中出现的次数。
方法一:BF
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.empty() || k < data[0] || k > data[data.size() - 1])
return 0;
int cnt = 0;
for(int i = 0; i < data.size(); ++i)
{
if(data[i] == k)
cnt ++;
}
return cnt;
}
};
方法二:二分查找第一个和最后一个k
class Solution {
public:
int getFirstK(vector<int>& data ,int k, int left, int right)
{
if(left > right)
return -1;
int mid = left + (right - left)/2;
if(data[mid] == k)//中间数字为k,判断是否为第一个k
{
if(mid == left || data[mid - 1] != k )
return mid;
else
right = mid -1;//前一个数字仍为k,则第一个k在前半段
}
else if(data[mid] > k)
right = mid - 1;
else
left = mid + 1;
return getFirstK(data, k, left, right);
}
int getLastK(vector<int>& data ,int k, int left, int right)
{
if(left > right)//不包含k返回-1
return -1;
int mid = left + (right - left)/2;
if(data[mid] == k)//中间数字为k,判断是否为最后一个k
{
if(mid == right || data[mid + 1] != k)
return mid;
else
left = mid + 1;//后一个数字仍为k,则最后一个k在后半段
}
else if(data[mid] > k)
right = mid - 1;
else
left = mid + 1;
return getLastK(data, k, left, right);
}
int GetNumberOfK(vector<int> data ,int k) {
if(data.empty())
return 0;
int cnt = 0;
int first = getFirstK(data, k, 0, data.size() - 1);
int last = getLastK(data, k, 0, data.size() - 1);
if(first > -1 && last > -1)
cnt = last - first + 1;
return cnt;
}
};
方法三:因为data中都是整数,所以可以稍微变一下,不是搜索k的两个位置,而是搜索k-0.5和k+0.5,这两个数应该插入的位置,然后相减即可。
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
return biSearch(data, k+0.5) - biSearch(data, k-0.5) ;
}
private:
int biSearch(const vector<int> & data, double num){
int s = 0, e = data.size()-1;
while(s <= e){
int mid = (e - s)/2 + s;
if(data[mid] < num)
s = mid + 1;
else if(data[mid] > num)
e = mid - 1;
}
return s;
}
};