剑指offer-37. 数字在排序数组中出现的次数

题目描述

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

解题思路：

因为数组有序，可用二分查找法，找到第一个k之后，用双指针向两边扩展

代码实现：

class Solution {
    /*
    因为数组有序，可用二分查找法，找到第一个k之后，用双指针向两边扩展
    */
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size() == 1 && data[0] == k){
            return 1;
        }
        int count = 0;
        int low = 0,high = data.size()-1;
        while(low < high){
            int mid = (low+high) / 2;
            if(data[mid] == k){
                //找到第一个k之后，用双指针向两边扩展
                count++;
                int up = mid+1;
                int down = mid-1;
                while(data[up]==k){
                    count++;
                    up++;
                }
                while(data[down]==k){
                    count++;
                    down--;
                }
                return count;
            }else if(data[mid] > k){
                high = mid - 1;
            }else{
                low = mid + 1;
            }
        }
        return 0;
    }
};

 效率：