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

最新推荐文章于 2022-11-13 17:16:41 发布

chasein

最新推荐文章于 2022-11-13 17:16:41 发布

阅读量112

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 文章标签：二分查找

本文链接：https://blog.youkuaiyun.com/weixin_40804971/article/details/85058102

剑指offer 专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种使用二分查找算法确定有序数组中特定元素出现次数的方法。通过两次调用二分查找，分别定位目标元素的首个和末位，从而计算其出现频率。示例代码展示了如何在给定数组中查找并计算指定数字5的出现次数。

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

#include<iostream>
#include<vector>
using namespace std;
int binarySearch(vector<int> data, int k, int low, int high){
    while(low <= high){  //必须加等号，否则此例中5将找不到
        int mid = (low + high) / 2;
        if(data[mid] == k)
            return mid;
        else if(data[mid] > k)
            high = mid - 1;
        else
            low = mid + 1;
    }

    return -1;  //没有找到
}
int GetNumberOfK(vector<int> data ,int k) {
     int size = data.size();
     int ret = binarySearch(data, k, 0, size - 1);
     if(ret == -1)
        return 0;
     int retL = ret;
     int retR = ret;
     if(ret >= 0 && ret <= size - 1){
        while(retL > 0 && data[retL - 1] == k){
            retL = binarySearch(data, k, 0, ret - 1);
        }

        while(retR < size - 1 && data[retR + 1] == k){
            retR = binarySearch(data, k, retR + 1, size - 1);
        }
     }
    return retR - retL + 1;
}

int main(){
	vector<int> v{2, 3, 3, 3, 5};
	//int ret = GetNumberOfK(v, 1);
	//int ret = GetNumberOfK(v, 3);
	//int ret = GetNumberOfK(v, 2);
	int ret = GetNumberOfK(v, 5);
    cout<<ret;
	return 0;
}