剑指offer 数字在排序数组中出现的次数

题目描述

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

思路：看到排序数组，首先考虑二分查找。我们找到数字在排序数组中最先出现和最后出现的下标，即得到次数。

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int len = data.size();
        int left = 0, right = len - 1, l, r;
        int mid;
        // 先找最左边的index，如果data[mid] == k,我们也将right = mid - 1,
        // 因为我们只要最左边的index，下面找最右边的index也是同理。
        while (left <= right) {
            mid = (right - left) / 2 + left;
            if (data[mid] >= k) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        l = left;
        right = len - 1;
        while (left <= right) {
            mid = (right - left) / 2 + left;
            if (data[mid] <= k) {
                left = mid + 1;
            } else {
                right = mid - 1;
            } 
        }
        r = right;
        return (r - l + 1);
    }
};