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

  • 问题描述
    统计一个数字在排序数组中出现的次数。
    class Solution {
    public:
    int GetNumberOfK(vector data ,int k) {

    }
    };

  • 分析
    最笨最耗时的方法,从头遍历,如果等于k,开始计数,不等于k时返回计数结果。但是这样时间复杂度为o(n),比较麻烦。
    比较聪明的方法1:因为排序数组,所以二分法找到第一个k,记下下标i,然后同样方法找到最后一个k,记下下标j,最后返回j-i+1。但是这样写了两个辅助方法后,又比较麻烦,提交后超时了。这个时候我们其实可以想到完全没必要找两个k的位置,一个头或者一个尾就足够了。所以有了比较聪明的方法2:
    如果找的是头那个k的下标,就向后查找,等于k计数器加1,不等就返回计数器的值。找的是尾的那个k的下标时同理。
    当然这个方法2也可以用二分法直接找k,随便找到哪一个,然后向左右遍历,找到最左最右端的k下标,然后再减。
    所以就有很多种代码可以达到这个目的,下面只列最后一种代码。

  • 代码

class Solution {
public:
    int GetK(vector<int>data,int k){
        int size=data.size();

        if(data.empty() || size==0){
            return-1;
        }
        int first=0;
        int end=size-1;
        int middle;

        while(first<=end){
             middle=(first+end)/2;
            if(data[middle]1;
            }else if(data[middle] >k){
                end=middle-1;
            }else{
                return middle;
            }
        }
        return -1;

    }
    int GetNumberOfK(vector<int> data ,int k) {
        int position=GetK(data,k);
        if(position==-1){
            return 0;
        }
        int i=position;
        int j=position;
        int size=data.size();
        while(i>=0 && data[i]==k)
            i--;
        while(jreturn j-i-1;

    }
};

你可能感兴趣的:(剑指offer)