IMWeb提升营Day7 | 训练题37：数字在排序数组中出现的次数

题目描述

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

思路一：暴力

/*
 * 思路一：暴力破解，没想到这也能过，看来测试用例也没有太大嘛
 */
class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int cnt = 0;
        for(int i = 0; i < data.size(); i++){
            if(data[i] == k) cnt++;
        }
        return cnt;
    }
};

思路二

/*
 * 思路二：既然是有序数组，当然要找效率更高的算法啦，必然是二分法
 */
class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int length = data.size();
        //因为二分法必须要数量在2以及以上，所以先处理特殊的情况
        if(length == 0) return 0;
        if(length == 1 && data[0] == k) return 1;
        int cnt = 0;
        int index = -1, indexback;
        int left = 0, right = data.size()-1;
        //先找到相等的那个数的索引，题目没明说，但是这里是按照升序数组来处理的
        while(left < right){
             int mid = (left + right) / 2;
            if(data[mid] == k){
                index = mid;
                break;
            }else if(data[mid] > k){
                right = mid - 1;
            }else{
                left = mid + 1;
            }
        }
        //找到索引之后，往前后两个方向找相同数字，累计数量，不相同即可停止寻找
        if(index >= 0){
            indexback = index - 1;
            while(indexback >= 0){
                if(data[indexback] == k) cnt++;
                else break;
                indexback--;
            }
            while(index < length){
                if(data[index] == k) cnt++;
                else break;
                index++;
            }
        }
        return cnt;
    }
};