题目描述
统计一个数字在排序数组中出现的次数。
分析:数组有序,采用二分查找无疑
两种方法,时间复杂度差不多,都是利用二分查找,不过统计k出现的次数有所不同而已
方法1:二分查找k,找到任意一个k的下标index,index向两边扩展即可
方法2:二分查找k+0.5和k-0.5的插入位置index1和index2,因为数组元素都是整数,index1-index2就是k出现的次数
时间复杂度:O(log N)
方法1:
int GetNumberOfK(vector<int> v ,int k) { if(v.size()==0) return 0; //查找任意一个k的位置index int low=0; int high=v.size()-1; int index=-1; while(low<=high) { int mid=(low+high)/2; if(v[mid]==k) { index=mid; break; } else if(v[mid]>k) { high=mid-1; } else if(v[mid]<k) { low=mid+1; } } if(index==-1) return 0; //index向两边扩展 int c=0; for(int i=index-1; i>=0; i--) { if(v[i]==k) c++; else break; } int n=v.size(); for(int i=index+1; i<n; i++) { if(v[i]==k) c++; else break; } return c+1; }
方法2:
//查找k的插入位置 int f(vector<int> &v,double k) { int low=0; int high=v.size()-1; while(low<=high) { int mid=(low+high)/2; if(v[mid]>k) { high=mid-1; }else if(v[mid]<k) { low=mid+1; } } return low; } int GetNumberOfK(vector<int> v ,int k) { if(v.size()==0) return 0; int x=f(v,k+0.5); int y=f(v,k-0.5); if(x==-1||y==-1) return 0; //k+0.5和k+0.5的插入位置相减就是k出现次数 return x-y; }
原文地址:https://www.cnblogs.com/yinbiao/p/11576729.html
时间: 2024-10-11 06:42:30