题目描述
统计一个数字在排序数组中出现的次数。
思路:在排序数组中出现次数,实际上就是查找问题。所以可以二分法。
PS:看到这题,我直接写了个map,直接映射查找,时间复杂度肯定不小,代码我也贴出来了。
//1.利用map
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
if(data.empty())
return 0;
int len = data.size();
map<int,int>ans;
for(int i =0;i<len;i++)
ans[data[i]]+=1;
map<int,int>::iterator itr;
itr = ans.find(k);
return itr->second;
}
};
对于二分法,有个巧妙的方法,利用查找周边的两个值find-0.5,和find+0.5,虽然找不到,但是可以找到最接近的位置,然后相减就可以了。
//2.二分法
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int a = MySearch(data,k-0.5);
int b = MySearch(data,k+0.5);
return b-a;
}
int MySearch(vector<int> data ,float k) {
int len = data.size();
int left = 0; int right =len-1;
int mid;
int num = 0;
while(left<=right)
{
mid = (left+right)/2;
if(k > data[mid])
left = mid+1;
else if(k < data[mid])
right = mid-1;
}
return right;
}
};