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

题目描述

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

思路

二分，当data[mid]==k的时候，在往左和右找等于K的，计数，计数的时候不要忘了data[mid]也，不要多算一次data[mid]，还有就是注意data只有一个元素的时候

code

class Solution {
public:

    int GetNumberOfK(vector<int> data ,int k) {
        int left=0;
        int right=data.size()-1;
        int mid=(left+right)/2;
        int res=0;
        while(left<=right){
            while(left<=right&&data[mid]>k){
                right=mid-1;
                mid=(left+right)/2;
            }
            while(left<=right&&data[mid]<k){
                left=mid+1;
                mid=(left+right)/2;
            }
            if(data[mid]==k){
                int temp=mid+1;
                while(mid>=0&&data[mid]==k){
                    res++;
                    mid--;
                }
                while(temp<data.size()&&data[temp]==k){
                    res++;
                    temp++;
                }
                return res;
            }
        }
        return res;
    }
};