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

最新推荐文章于 2019-03-17 02:35:26 发布

FBI-PC

最新推荐文章于 2019-03-17 02:35:26 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏：二分与三分剑指offer

本文链接：https://blog.youkuaiyun.com/u014303647/article/details/81636705

剑指offer 同时被 2 个专栏收录

54 篇文章

订阅专栏

二分与三分

9 篇文章

订阅专栏

本文介绍了一种高效的方法来统计一个数字在已排序数组中出现的次数。通过使用C++ STL中的lower_bound和upper_bound函数，可以快速定位目标数字的范围，从而计算其出现次数。此外，还提供了这两个函数的自实现版本，帮助读者深入理解其工作原理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

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

解法一：其实很简单，两行代码就搞定，就是用C++ stl里面的lower_bound和upper_bound，lower_bound是找出不小于即大于等于的第一个数的下标 upper_bound是找出大于的第一个数的下标

代码：

class Solution 
{
public:
    int GetNumberOfK(vector<int> data ,int k)
    {
        int s1 = lower_bound(data.begin(),data.end(),k)-data.begin();
        int s2 = upper_bound(data.begin(),data.end(),k)-data.begin();
        //cout<<s1<<" "<<s2<<endl;
        return s2-s1;
    }
};

解法二：想刨根问底的话，就要知道lower_bound和upper_bound是怎么实现的，参考STL源码剖析，我们可以写出简化版本

class Solution
{
public:
    int my_lower_bound(vector<int> data, int key)
    {
        int half;
        int len = data.size();
        int mid;
        int first = 0;
        while (len > 0)
        {
            half = len >> 1;
            mid = first + half;
            //in the right part
            if (data[mid] < key)
            {
                first = mid + 1;
                //因为first=mid+1,所以这里的len需要在减去half的基础之上再减去1
                len = len - half - 1;
            } else
            {
                //in the left part
                len = half;
            }
        }
        return first;
    }


    int my_upper_bound(vector<int> data, int key)
    {
        int mid;
        int first = 0;
        int len = data.size();
        int half;
        while (len > 0)
        {
            half = len >> 1;
            mid = half + first;
            if (data[mid] > key)
            {//in the left part
                len = half;
            } else
            {//if arr[mid]<= key ,in the right part
                first = mid + 1;
                len = len - half - 1;
            }
        }
        return first;
    }


    int GetNumberOfK(vector<int> data ,int k)
    {
        int s1 = my_lower_bound(data,k);
        int s2 = my_upper_bound(data,k);
        cout<<s1<<" "<<s2<<endl;
        return s2-s1;
    }
};