题目描述
统计一个数字在升序数组中出现的次数。
思路
思路1:暴力
暴力也是要有讲究的,直接遍历一遍统计当然是可以的,但是没有用到有序的特征,因此可以在遍历的时候加一个检测。
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if (data.size()==0) return 0;
int n=data.size();
int i=0,ret=0;
while (i<n&&data[i]<=k)//一旦遍历到比k大就赶紧停下来,不做后续的检测了
{
if (data[i]==k)
ret++;
i++;
}
return ret;
}
};
思路2: 二分法
在一个有序数组中寻找指定的值当然要想到二分法
上界定义为第一个比目标k大的数
下界定义为第一个k或者第一个比k大的数(当k不存在的时候)
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if (data.size()==0) return 0;
//寻找下界
int l=0,r=data.size();
while(l<r)
{
int mid=l+(r-l)/2;
if (data[mid]<k)
l=mid+1;
else
r=mid;
}
int l_bound=r;
//寻找上界
l=0,r=data.size();
while(l<r)
{
int mid=l+(r-l)/2;
if (data[mid]<=k)
l=mid+1;
else
r=mid;
}
int r_bound=l;
return r_bound-l_bound;
}
};
还能利用标准库(标准库真香!)
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int l = lower_bound(data.begin(), data.end(), k);
int r = upper_bound(data.begin(), data.end(), k);
return r - l;
}
};
其中
int l = lower_bound(data.begin(), data.end(), k);
int r = upper_bound(data.begin(), data.end(), k);
这个找上界和下界的函数底层就是用二分查找实现的
还有更过分的:
return count(data.begin(), data.end(), k);
一行代码实现,但是这个的底层是直接遍历实现的,效率并没有二分高