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

最新推荐文章于 2022-07-28 23:54:26 发布

转载最新推荐文章于 2022-07-28 23:54:26 发布 · 196 阅读

收录于

LeetCode 专栏收录该内容

39 篇文章

订阅专栏

本文介绍了一种优化的二分查找算法，通过查找k+0.5和k-0.5的位置来快速确定目标数字在有序数组中出现的次数，这种方法在面试中不易出错且效率高。

查找一个数组中的某个数字的个数
正常解法是：二分位置找前后端点，然后相减
眼前一亮的解法,并且不容易面试手写出错的解法如下：
二分查找k+0.5 和 k-0.5的位置，然后位置相减

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        return differ(data,k + 0.5) - differ(data,k - 0.5);
    }
    int differ(vector<int> data ,double s) {
        int mid = 0, l = 0, r = data.size();
        while(l <= r) {
            mid = (l + r)/2;
            if(data[mid] > s ) r = mid - 1;
            if(data[mid] < s ) l = mid + 1;            
        }
        return l;
    }
};
int main(){
   Solution s;
   vector<int> a;
   a.clear();
   a.push_back(1);
   a.push_back(2);
   a.push_back(2);
   a.push_back(2);
   a.push_back(2);
   a.push_back(2);
   a.push_back(4);
   a.push_back(4);
   a.push_back(5);
   cout<<s.GetNumberOfK(a,3)<<endl;
}