剑指offer-JZ37-数字在升序数组中出现的次数

题目描述

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

 

思路：

排序数组查找元素，则用二分法，在二分基础上第一次找到重复元素左边界，第二次找到重复元素右边界。查找边界的时候注意递归结束条件和找到边界时的情况。查找左边界时，当中间元素center等于要查找的元素k时，判断center前一个元素是否也等于k，等于k则说明该位置不是边界，需要继续在左半部分查找；当center前一个元素不等于k或者该位置为0即数组第一个位置时，则说明查找到了左边界。右边界同理

C++

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.empty())
            return 0;
        int res = 0;
        int n = data.size();
        int l = BinarySearchLeft(data, k, 0, n - 1);
        if(l == -1 )
            return res;
        int r = BinarySearchRight(data, k, 0, n - 1);
        res = r - l + 1;
        return res;
    }
    
    int BinarySearchLeft(vector<int> data, int k, int l, int r){
        int center = l + (r - l) / 2;
        if(l > r)
            return -1;
        if(data[center] == k){
            if((center >= 1 && data[center - 1] != k )|| center == 0)
                return center;
            else
                return BinarySearchLeft(data, k, l, center - 1);
        }
        else if(data[center] > k)
            return BinarySearchLeft(data, k, l, center - 1);
        else
            return BinarySearchLeft(data, k, center + 1, r);
    }
    
    int BinarySearchRight(vector<int> data, int k, int l, int r){
        int center = l + (r - l) / 2;
        if(l > r)
            return -1;
        if(data[center] == k){
            if((center <= data.size()-2 && data[center + 1] != k) || center == data.size() - 1)
               return center;
            else
                return BinarySearchRight(data, k, center + 1, r);
        }
        else if(data[center] > k)
            return BinarySearchRight(data, k, l, center - 1);
        else
            return BinarySearchRight(data, k, center + 1, r);
    }
};