题目:
统计一个数字在排序数组中出现的次数。
思路:
对二分查找进行改进,找到数字在数组中第一次出现和最后一次出现的位置,这样就得到它出现的次数。
以找第一次出现的位置为例:如果mid元素大于k,则在前半段找;如果小于k,则在后半段找;如果等于k,则要看mid的前一个元素是不是k,如果是,则在前半段找,如果不是,则这就是第一次出现的位置。
时间复杂度O(logn)。
代码:
class Solution { public: int GetNumberOfK(vector<int> data ,int k) { if(data.size()<=0) return 0;//OMG!这里也是应该输出0 int first=GetFirstK(data,k,0,data.size()-1); int last=GetLastK(data,k,0,data.size()-1); if(first>=0 && last>=0) { return last-first+1; } return 0;//OMG!这里应该是0啊 } private: int GetFirstK(const vector<int> &data, int k, int begin, int end) { if(begin<0 || end>=data.size() || begin>end) return -1; int res=-1; int mid=begin+(end-begin)/2; if(data[mid]>k) res=GetFirstK(data,k,begin,mid-1); else if(data[mid]<k) res=GetFirstK(data,k,mid+1,end); else { if(mid==0 || (mid>0 && data[mid-1]!=k)) res=mid; else res=GetFirstK(data,k,begin,mid-1); } return res; } int GetLastK(const vector<int> &data, int k, int begin, int end) { if(begin<0 || end>=data.size() || begin>end) return -1; int res=-1; int mid=begin+(end-begin)/2; if(data[mid]>k) res=GetLastK(data,k,begin,mid-1); else if(data[mid]<k) res=GetLastK(data,k,mid+1,end); else { if(mid==data.size()-1 || (mid<data.size()-1 && data[mid+1]!=k)) res=mid; else res=GetLastK(data,k,mid+1,end); } return res;//!!!忘了这一句,找了好长时间错误!!! } };