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

题目描述:统计数字在排序数组中出现的次数,例如输入数组{1,2,3,3,3,3,4,5,7,8}和数字3,则输出4,因为数字3在数组中出现了4次。

解:此题最简单做法是扫描数组,每当碰到与数字一样的数组元素时,次数+1。这种做法的时间复杂度为O(N),而且没有用上数组有序这个性质。

另一种做法是用二分法找到对应的数组元素下标,然后分别往数组的左边和右边扫描,找到第一个与最后一个元素。但是这种做法的时间的时间复杂度依然为O(N)。

比较好的方法是,用二分法查找出数字的起始位置和末尾位置。这样的话时间复杂度为O(logN),代码如下:

class Solution {
public:
    int GetNumberOfK(vector<int> data ,int k) {
        int Size = data.size();
        if(Size==0) return 0;
        if(data[0]>k||data[Size-1]<k) return 0;
        int FirstK = GetFirstK(data,k);
        int LastK = GetLastK(data,k);
        if(FirstK!=-1&&LastK!=-1) return LastK-FirstK+1;
        else return 0;
    }
    int GetFirstK(vector<int> data, int k){
        int Size = data.size();
        int Left = 0, Right = Size-1, IndexK = -1;
        if(data[Left]==k) return Left;
        int Mid;
        while(Left<=Right){
            Mid = (Left+Right)/2;
            if(data[Mid]==k) {
                if(data[Mid-1]<k) return Mid;
                else Right = Mid-1;
            }
            else if(data[Mid]<k) Left = Mid+1;
            else Right = Mid-1;
            
        }
        return -1;
    }
    int GetLastK(vector<int> data, int k){
        int Size = data.size();
        int Left = 0, Right = Size-1, IndexK = -1;
        if(data[Right]==k) return Right;
        int Mid;
        while(Left<=Right){
            Mid = (Left+Right)/2;
            if(data[Mid]==k){
                if(data[Mid+1]>k) return Mid;
                else Left = Mid+1;
            }
            else if(data[Mid]>k) Right = Mid-1;
            else Left = Mid+1;
            }
        return -1;
    }
};

主函数是GetNumberOfK。函数首先要判断输入数组的有效性,如数组是否空,第一个元素是否大于K,最后一个元素是否小于K。这个判断非常重要,否则可能会导致数组溢出。之后用GetFirstK和GetLastK函数获取数组第一个K和最后一个K出现的下标。两者之差即为K出现的次数。两个函数功能类似,因此只解析GetFirstK。

函数首先判断一下第一个元素是否等于K,如果是的话可以直接返回第一个元素下标,同时也避免下方循环中算data[Mid-1]时数组溢出。
循环中不断比较中间元素和K。如果中间元素大于K,则K在数组前半段。若中间元素大于K,则K在数组前半段。如果中间元素等于K,则比较一下前一个元素是否小于K,如果是,则此元素为K第一次出现的位置,直接返回。否则元素K第一次出现的位置一定在数组的前半段。

题目来源https://www.nowcoder.com/practice/70610bf967994b22bb1c26f9ae901fa2?tpId=13&tqId=11190&tPage=1&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值