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

最新推荐文章于 2024-10-23 23:07:46 发布

原创最新推荐文章于 2024-10-23 23:07:46 发布 · 129 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二分查找

剑指offer 专栏收录该内容

22 篇文章

订阅专栏

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

#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;
}