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

题目描述：

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

分析：

 二分变形。二分查找最左边和最右边k的位置，然后相减加一就是结果。

代码：

class Solution {
public:
    int GetNumberOfK(vector<int> data, int k) {
        int dataSize = data.size();
        if(dataSize == 0) return 0;
        int firstK = GetPositionOfFirstK(data, k, 0, dataSize - 1);
        if(firstK == -1) return 0;  // k不存在
        int lastK = GetPositionOfLastK(data, k, 0, dataSize - 1);
        return lastK - firstK + 1;
    }
    int GetPositionOfFirstK(vector<int> data, int k, int left, int right) { // 二分找最左的k的位置
        int mid = (left + right) >> 1;
        while(left < right) {
            if(data[mid] > k) {
                right = mid - 1;
            } else if(data[mid] < k) {
                left = mid + 1;
            } else {
                if(data[left] != k) left++;
                else return left;
                right = mid;
            }
            mid = (left + right) >> 1;
        }
        return data[mid] == k ? mid : -1;
    }
    int GetPositionOfLastK(vector<int> data, int k, int left, int right) {  // 二分找最右的k的位置
        int mid = (left + right) >> 1;
        while(left < right) {
            if(data[mid] < k) {
                left = mid + 1;
            } else if(data[mid] > k) {
                right = mid - 1;
            } else {
                if(data[right] != k) right--;
                else return right;
                left = mid;
            }
            mid = (left + right) >> 1;
        }
        return data[mid] == k ? mid : -1;
    }
};