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

Posted 2022-08-09 zhudingtop

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

【思路】首先要清楚这种看似简单的题目，使用直接遍历是可以，但一般不得分，由于题目给出了排序数组，对于排序数组来说，常用的搜索查找方式为二分查找（binary search）。这里有个巧妙的方法，我们并不是去搜索k这个数，而是去搜索k-0.5和k+0.5这两个小数，进而返回待插入的位置！

比如：1 2 2 2 3 4 且k = 2
则，k+0.5会返回3的索引即4，而k-0.5会返回第一个2的索引即1，两者相减得3，即为最后的结果！二分代码如下，只返回begin位置！

class Solution 
public:
    int GetNumberOfK(vector<int> data ,int k) 
        return biSearch(data, k+0.5) - biSearch(data, k-0.5);
    
private:
    int biSearch(vector<int>& data, double num)
        int begin = 0, end = data.size() - 1;
        while(begin <= end)
            int mid = begin + (end - begin) / 2;
            if(data[mid] < num)
                begin = mid + 1;
            else if(data[mid] > num)
                end = mid - 1;
        
        return begin;
    
;

另外一个思路直接使用STL中的库函数equal_range来获取与某一个值相等的上下边界，十分好用的！

class Solution 
public:
    int GetNumberOfK(vector<int> data ,int k) 
        auto res = equal_range(data.begin(), data.end(), k);
        return res.second - res.first;
    
;