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

最新推荐文章于 2020-09-09 17:02:24 发布

原创最新推荐文章于 2020-09-09 17:02:24 发布 · 194 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

牛客网（剑指offer）专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种高效算法来统计排序数组中特定数字的出现次数。通过改进的二分查找方法快速定位目标元素，并统计其出现频次，显著提高了算法效率。

题目描述

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

思路：可以使用暴力枚举，时间复杂度O(N)，但肯定不是最优解

思路二：因为数组data是排好序的，可以利用这一信息，采用改进的二分查找，找到值为Key的最小索引下标index，然后往右搜索，统计出现次数，代码如下：

代码：

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int res = 1;
        int n = data.size();
        if(n == 0) return 0;
        
        int left = 0;
        int right = n-1;
        int mid;
        int cnt = 0;
        while(left + 1 < right) {
            mid = (left + right) / 2;
            if(data[mid] >= k) {
                right = mid;
            } else {
                left = mid + 1;
            }
        }
        if(data[left] == k) cnt = left;
        else if(data[right] == k) cnt = right;
        else return 0;
        
        while(++cnt < n && data[cnt] == k) {
            ++res;
        }
        return res;
    }
};