练习：数字在排序数组中出现的次数

数字在排序数组中出现的次数

一、题目描述

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

二、我的思路

1、使用二分查找法找出数组中值为k的任意一个数，并记录其位置pos；
2、在pos前后分别扩展，找到pos前后所有值为k的数，记录个数；

三、我的代码

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size() == 0){
            return 0;
        }
        int pos = 0, res = 0;
        int low = 0, high = data.size()-1;
        while(low <= high){
            int middle = (low + high)/2;
            if(data[middle] == k){
                pos = middle;
                res++;
                break;
            }
            else if(k > data[middle]){
                low = middle+1;
            }
            else if(k < data[middle]){
                high = middle-1;
            }
        }
        if(res != 0){
            int temp;
            temp = pos;
            while(pos >= 0 && data[--temp] == k){
                res++;
            }
            temp = pos;
            while(pos <= data.size()-1 && data[++temp] == k){
                res++;
            }
        }
        return res;
    }
};