题目描述
统计一个数字在排序数组中出现的次数。
思路1:双子针法,头指针从头向后开始遍历查找,找到即退出到当前位置i,尾指针从尾部开始向前查找,找到即退出到当前位置j,最后两个指针的位置相减,再加1,得到的就是该查找的数字在数组中出现的次数。
思路2:顺序数组可以用二分查找,上面这种做法的时间复杂度为O(n)O(n),我们没有充分利用有序这个条件,有序序列的查找可以用二分来做,查找第一个数字用二分求下界;查找第二个数字用二分求上界。
这样一做,时间复杂度就是O(logn)O(logn)了。
思路1代码:
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int i = 0,j = data.size()-1;
while(i<data.size())
{
if(data[i] == k) break;
i++;
if(i == data.size()) return 0;//找了一遍都没找打就返回0
}
while(j>=0 && j>i)
{
if(data[j] == k) break;
j--;
}
return j-i+1;
}
};
思路2代码:
class Solution {
public:
int GetNumberOfK(vector<int> data ,int k) {
int pre, last, l, r;
for (l = 0, r = data.size() ;l < r; ) {
int mid = (l + r) / 2;
if (data[mid] >= k)
r = mid;
else
l = mid + 1;
}
pre = r;
for (l = -1, r = (int)data.size() - 1; l < r; ) {
int mid = (l + r + 1) / 2;
if (data[mid] <= k)
l = mid;
else
r = mid - 1;
}
last = l;
return last - pre + 1;
}
};