【剑指offer】面试题38:数字在排序数组中出现的次数

时间:2023-03-09 01:26:12
【剑指offer】面试题38:数字在排序数组中出现的次数

题目:

统计一个数字在排序数组中出现的次数。

思路:

对二分查找进行改进,找到数字在数组中第一次出现和最后一次出现的位置,这样就得到它出现的次数。

以找第一次出现的位置为例:如果mid元素大于k,则在前半段找;如果小于k,则在后半段找;如果等于k,则要看mid的前一个元素是不是k,如果是,则在前半段找,如果不是,则这就是第一次出现的位置。

时间复杂度O(logn)。

代码:

class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.size()<=) return ;//OMG!这里也是应该输出0 int first=GetFirstK(data,k,,data.size()-);
int last=GetLastK(data,k,,data.size()-);
if(first>= && last>=)
{
return last-first+;
} return ;//OMG!这里应该是0啊
}
private:
int GetFirstK(const vector<int> &data, int k, int begin, int end)
{
if(begin< || end>=data.size() || begin>end) return -; int res=-;
int mid=begin+(end-begin)/;
if(data[mid]>k)
res=GetFirstK(data,k,begin,mid-);
else if(data[mid]<k)
res=GetFirstK(data,k,mid+,end);
else
{
if(mid== || (mid> && data[mid-]!=k))
res=mid;
else
res=GetFirstK(data,k,begin,mid-);
} return res;
} int GetLastK(const vector<int> &data, int k, int begin, int end)
{
if(begin< || end>=data.size() || begin>end) return -; int res=-;
int mid=begin+(end-begin)/;
if(data[mid]>k)
res=GetLastK(data,k,begin,mid-);
else if(data[mid]<k)
res=GetLastK(data,k,mid+,end);
else
{
if(mid==data.size()- || (mid<data.size()- && data[mid+]!=k))
res=mid;
else
res=GetLastK(data,k,mid+,end);
}
return res;//!!!忘了这一句,找了好长时间错误!!!
}
};