最大值减去最小值小于或等于num的子数组数量

博客围绕给定数组arr和整数num,求解满足子数组中最大值减最小值小于等于num的子数组数量展开。介绍了基本思路,利用两个双端队列qmin和qmax维护最值更新,通过移动i和j确定子数组范围,时间复杂度为O(N)。

题目:给定数组arr和整数num,共返回有多少个子数组满足如下情况:子数组中的最大值减去最小值小于或等于num。
要求,时间复杂度O(N)。

基本思路:

首先明确两点:
  
  1、如果子数组arr[i…j]满足条件,那么arr[i…j]中的子数组一定也满足条件。
  2、如果子数组arr[i…j]不满足条件,那么包含arr[i…j]的子数组一定也不满足条件。

使用两个双端队列qmin和qmax,qmin用来维护子数组arr[i…j]的最小值更新,qmax用来维护子数组arr[i…j]的最大值更新。队头表示的就是子数组arr[i…j]的最小(大)值。生成两个整型变量i和j,用于表示子数组的范围,即arr[i…j]。整型变量res表示所有满足条件的子数组数量。初始时,i、j、res都为0。

令j不断向右移动,表示arr[i…j]一直向右扩张,并不断更新qmin和qmax。一旦出现arr[i…j]不满足条件的情况,扩张停止,此时arr[i…j-1]、arr[i…j-2]、arr[i…j-3]…arr[i…i]一定满足条件。即所有必须以arr[i]开头的满足条件的子数组数量为j - i。所以令 res += j - i。

向右扩张停止后,令i向右移动一个单位,表示开始考虑以arr[i+1]开头的满足条件的子数组数量,更新qmin、qmax。接下来的过程和上述一样。

所有的下标值最多进qmax和qmin一次,出qmax和qmin一次。i和j的值也不断增加,并且从来不减小,所以整个时间复杂度为O(N)

def getNum(L,num):
    if L == None or len(L) < 1 or num <0 :
        return

    qmin = []
    qmax = []

    i = 0
    j = 0

    res = 0

    while i < len(L):
        while j < len(L):
            while len(qmin)!=0 and L[qmin[-1]] >= L[j]:
                qmin.pop()

            qmin.append(j)

            while len(qmax)!=0 and L[qmax[-1]] <= L[j]:
                qmax.pop()

            qmax.append(j)

            if L[qmax[0]] - L[qmin[0]] > num:
                break
            j +=1
        res += j-1

        if qmin[0] == i:
            qmin.pop(0)
        if qmax[0] == i:
            qmax.pop(0)

        i +=1

    return res


 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值