剑指offer 面试题53 数字在排序数组中出现的次数

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

tips: 一种方法，先通过二分查找找到这个数字，然后向左和向右遍历，计算这个数字出现的次数， 复杂度o(n)
另外一种方法，先找到第一个k和最后一个k的位置，二者相减，得出出现次数，
找第一个k和最后一个k的过程与二分查找的过程类似，总体的复杂度为o(log(n))

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size()==0) {
            return 0;
        } 
        int firstK=FindFirstK(data,k,0,data.size()-1);
        int lastK=FindLastK(data,k, 0,data.size()-1);
        if(firstK==-1&&lastK==-1) {
            return 0;
        } else if(firstK==-1||lastK==-1) {
            return 1;
        } else {
            return  lastK-firstK+1;
        }
    }
    int FindFirstK(vector<int>& data,int k,int start,int end) {
        if(start==end) {
            if(data[start]==k) {
                return start;
            } else {
                return -1;
            }
        } else if(start>end) {
            return -1;
        }
        int mid=(start+end)/2;
        if(data[mid]==k) {
            if(mid==start||data[mid-1]<k) {
                return mid;
            } else {
                return FindFirstK(data,k,start,mid);
            }
        } else if(data[mid]>k) {
            return FindFirstK(data,k,start,mid);
        } else {
            return FindFirstK(data,k,mid+1,end);
        }
    }
    int FindLastK(vector<int>& data,int k,int start,int end) {
        if(start==end) {
            if(data[start]==k) {
                return start;
            } else {
                return -1;
            }
        } else if(start>end) {
            return -1;
        }
        int mid=(start+end)/2;
        if(data[mid]==k) {
            if(mid==end||data[mid+1]>k) {
                return mid;
            } else {
                return FindLastK(data,k,mid+1,end);
            }
        } else if(data[mid]>k) {
            return FindLastK(data,k,start,mid);
        } else {
            return FindLastK(data,k,mid+1,end);
        }
    }
};