[剑指Offer] 37.数字在排序数组中出现的次数

题目描述

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

【思路】因为是排序数组，所以可以用二分法搜索到要查找的值在数组中的一个位置，接着向两侧扫描，遇到不等的就停止。

class Solution {
public:
    int getIndexbyDivision(vector<int> data,int k){
        int left = 0,right = data.size() - 1;
        int middle = left + (right - left)/2;
        while(left <= right){
            if(k > data[middle])
                left = middle + 1;
            else if(k < data[middle])
                right = middle - 1;
            else 
                return middle;
            middle = left + (right - left)/2;
        }
        return -1;
    }
    int GetNumberOfK(vector<int> data ,int k) {
        int index = getIndexbyDivision(data,k);
        if(index == -1) return 0;
        int num = 1;
        for(int i = index - 1;i >= 0 && data[i] == k;i --)    num++;
        for(int j = index + 1;j < data.size() && data[j] == k;j ++)    num++;
        return num;
    }
};