剑指offer——56在排序数组中查找数字

题目描述

统计一个数字在排序数组中出现的次数。

 

题解：

　　使用二分法找到数k然后向前找到第一个k，向后找到最后一个k，即可知道有几个k了

　　但一旦n个数都是k时，这个方法跟从头遍历没区别，都是O(N)的复杂度

　　可以再次利用二分法，在第一次找到k的左半部分使用二分法找到不再出现k的位置，其右半部份类似。

　　

class Solution01 {
public:
    int GetNumberOfK(vector<int> data, int k) {
        if (data.size() == 0)return 0;
        int L = 0, R = data.size() - 1, M;
        while (L <= R)
        {
            M = (R - L) / 2 + L;
            if (data[M] == k)break;
            else if (data[M] > k)R = M - 1;
            else L = M + 1;
        }
        if (L > R)return 0;
        L = R = M;
        int res = 0;
        while (L >= 0 && data[L] == k) {
            res++; --L;
        }
        while (R<data.size() && data[R] == k) {
            res++; ++R;
        }
        return res - 1;
    }
};

class Solution02 {
public:
    int GetNumberOfK(vector<int> data, int k) {
        if (data.size() == 0)return 0;
        int pM = find(data, 0, data.size() - 1, k);
        if (data[pM] != k)return 0;
        int pL = pM, pR = pM;
        while (pL !=-1 && data[pL] == k) pL = find(data, 0, pL - 1, k);
        while (pR != -1 && data[pR] == k) pR = find(data, pR + 1, data.size() - 1, k);
        return (pR == -1 ? data.size() : pR) - pL - 1;
    }
    int find(const vector<int>data, int L, int R, const int k)
    {
        int M = -1;
        while (L >= 0 && R < data.size() && L <= R)
        {
            M = (R - L) / 2 + L;
            if (data[M] == k)break;
            else if (data[M] > k)R = M - 1;
            else L = M + 1;
        }
        return M;
    }
};