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

题目描述

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

分析：排序数组找数字，当然想到用二分，用两次二分，分别找到第一个k和最后一个k，计算两者位置的差值返回。要注意数组中不包含这个数字的情况。

代码：

class Solution {
public:
    int GetFirstk(vector<int> data,int k,int low,int high){
        while(low<high){
            int mid=low+(high-low)/2;
            if(data[mid]==k){
                if(data[mid-1]!=k){
                    return mid;
                }
                high=mid-1;
            }
            else if(data[mid]>k){
                high=mid-1;
            }else{
                low=mid+1;
            }
            
        }
        if(low>=high&&data[low]==k) return low;
        return -1;
    }
    int GetLastk(vector<int> data,int k,int low,int high){
        while(low<high){
            int mid=low+(high-low)/2;
            if(data[mid]==k){
                if(data[mid+1]!=k){
                    return mid;
                }
                low=mid+1;
            }
            else if(data[mid]>k){
                high=mid-1;
            }
            else{
                low=mid+1;
            }
        }
        if(low>=high&&data[low]==k) return low;
        return -1;
    }
    int GetNumberOfK(vector<int> data ,int k) {
        int n=data.size();
        if(n==0||k<data[0]||k>data[n-1]) return 0;
        int first=GetFirstk(data,k,0,n-1);
        int last=GetLastk(data,k,0,n-1);
        if(last!=-1&&first!=-1) return last-first+1;
        else return 0;
    }
};