码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2019-08-20 21:58:08      阅读:86      评论:0      收藏:0      [点我收藏+]

标签:color   equal   函数   span   cond   结果   代码   搜索   插入   

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

【思路】首先要清楚这种看似简单的题目,使用直接遍历是可以,但一般不得分,由于题目给出了排序数组,对于排序数组来说,常用的搜索查找方式为二分查找(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;
    }
};

 

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

标签:color   equal   函数   span   cond   结果   代码   搜索   插入   

原文地址:https://www.cnblogs.com/zhudingtop/p/11385845.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!