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

潭溪熹子

于 2018-11-16 20:55:21 发布

阅读量248

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer 文章标签：剑指offer 基础算法二分查找

本文链接：https://blog.youkuaiyun.com/weixin_40271182/article/details/84145758

剑指offer 专栏收录该内容

66 篇文章

订阅专栏

通过二分查找法，有效地在有序数组中找到目标数字首次出现和最后一次出现的角标，从而计算出该数字在数组中出现的次数。步骤包括定位最前面和最后面的元素位置，最终结果为这两个角标之差加一。

/*
思路：
数组有序马上想到用二分查找。
用二分查找找到最前面元素和最后面元素的角标i，j。
结果就是：j-i+1。
*/

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        if(data.size()==0)
            return 0;
        int number = 0;
        int first = GetFirstIndex(data,k,0,data.size()-1);
        int last = GetLastIndex(data,k,0,data.size()-1);
        if(first>-1 && last>-1)
            number = last - first + 1;
        return number;
    }
    int GetFirstIndex(vector<int> data,int k,int start,int end){
        if(start > end)
            return -1;
        int mid = (start + end)/2;
        if(data[mid] == k){
            if(mid == start || data[mid-1] != k)
                return mid;
            else
                end = mid - 1;
        }else if(data[mid] > k)
            end = mid - 1;
        else 
            start = mid + 1;
        return GetFirstIndex(data,k,start,end);
    }
    
    int GetLastIndex(vector<int> data,int k,int start,int end){
        if(start > end)
            return -1;
        int mid = (start + end)/2;
        if(data[mid] == k){
            if(mid == end || data[mid+1] != k)
                return mid;
            else
                start = mid + 1;
        }else if(data[mid] > k)
            end = mid - 1;
        else 
            start = mid + 1;
        return GetLastIndex(data,k,start,end);
    }
    
};