题目描述
统计一个数字在排序数组中出现的次数。
分析:使用二分法找到第一个k和最后一个k出现的位置
1. 如果midData> k,那么第一个和最后一个k肯定在左半边;
2. 如果midData< k,那么第一个和最后一个k肯定在右半边;
3. 如果midData=k,那么需要判断这个k是不是第一个或最后一个k:
(1)如果midData左侧相邻的数是k,那么第一个k一定在左半边;
(2)如果midData左侧相邻的数不是k,那么第一个k就是中间数字;
(3)如果midData右侧相邻的数是k,那么最后一个k一定在右半边;
(4)如果midData右侧相邻的数不是k,那么最后一个k就是中间数字。
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int number=0;
int len=data.size();
if(len==0){
return 0;
}
else{
int first=GetFirstK(data,len,k,0,len-1);
int last=GetLastK(data,len,k,0,len-1);
if(first>-1&&last>-1){
number=last-first+1;
}
}
return number;
}
int GetFirstK(vector<int> data,int len,int k,int start,int end){
if(start>end){
return -1;
}
int mid=(start+end)/2;
int midData=data[mid];
if(midData==k){
if((mid>0&&data[mid-1]!=k)||mid==0){
return mid;
}
else{
end=mid-1;
}
}
else if(midData>k){
end=mid-1;
}
else{
start=mid+1;
}
return GetFirstK(data,len,k,start,end);
}
int GetLastK(vector<int> data,int len,int k,int start,int end){
if(start>end){
return -1;
}
int mid=(start+end)/2;
int midData=data[mid];
if(midData==k){
if((mid>0&&data[mid+1]!=k)||mid==0){
return mid;
}
else{
start=mid+1;
}
}
else if(midData>k){
end=mid-1;
}
else{
start=mid+1;
}
return GetLastK(data,len,k,start,end);
}
};