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

题目描述：

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

思路：

看到排序数组，首先想到折半查找。折半查找到数字k第一次和最后一次出现的位置，两者相减即为出现的次数。时间复杂度O(logn)。若书序查找，时间复杂度则为O(n)。

代码：

语言C++，已通过牛课网在线测试。

class Solution {
public:
    
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size() == 0) return 0;
        int first = getFirst(data, k);
        int last = getLast(data, k);
        return last - first + 1;
    }
    
    int getFirst(vector<int> data, int k){
        int start = 0, end = data.size() - 1;
        int mid = (start + end) >> 1;
        while(start <= end){
            if(data[mid] < k) start = mid + 1;
            else end = mid - 1;//=k的话 end左移，找到第一个
            mid = (start + end) >> 1;
        }
        return start;
    }
    int getLast(vector<int> data, int k){
        int start = 0, end = data.size() - 1;
        int mid = (start + end) >> 1;
        while(start <= end){
            if(data[mid] <= k) start = mid + 1;//=k的话 start右移 找到最后一个
            else end = mid - 1;
            mid = (start + end) >> 1;
        }
        return end;
    }
    
};