题目描述
统计一个数字在排序数组中出现的次数。
题解:
使用二分法找到数k然后向前找到第一个k,向后找到最后一个k,即可知道有几个k了
但一旦n个数都是k时,这个方法跟从头遍历没区别,都是O(N)的复杂度
可以再次利用二分法,在第一次找到k的左半部分使用二分法找到不再出现k的位置,其右半部份类似。
class Solution01 {
public:
int GetNumberOfK(vector<int> data, int k) {
if (data.size() == )return ;
int L = , R = data.size() - , M;
while (L <= R)
{
M = (R - L) / + L;
if (data[M] == k)break;
else if (data[M] > k)R = M - ;
else L = M + ;
}
if (L > R)return ;
L = R = M;
int res = ;
while (L >= && data[L] == k) {
res++; --L;
}
while (R<data.size() && data[R] == k) {
res++; ++R;
}
return res - ;
}
}; class Solution02 {
public:
int GetNumberOfK(vector<int> data, int k) {
if (data.size() == )return ;
int pM = find(data, , data.size() - , k);
if (data[pM] != k)return ;
int pL = pM, pR = pM;
while (pL !=- && data[pL] == k) pL = find(data, , pL - , k);
while (pR != - && data[pR] == k) pR = find(data, pR + , data.size() - , k);
return (pR == - ? data.size() : pR) - pL - ;
}
int find(const vector<int>data, int L, int R, const int k)
{
int M = -;
while (L >= && R < data.size() && L <= R)
{
M = (R - L) / + L;
if (data[M] == k)break;
else if (data[M] > k)R = M - ;
else L = M + ;
}
return M;
}
};