单调栈结构

题目:给定一个不含重复值的数组,找到每一个i位置左边和右边离i位置最近且值比arr[i]小的位置。返回所有位置相应信息

举例:

arr = {3,4,1,5,6,2,7}

返回数组:{{-1,2},{0,2},{-1,-1},{2,5},{3,5},{2,-1},{5,-1}}

进阶问题:给定一个可能含有重复值的数组arr,找到每一个i位置左边和右边离i位置最近且值比arr[i]小的位置。返回所有位置相应信息

要求:如果arr长度为N,实现原问题和进阶问题的解法,时间复杂度都达到O(N)

思路:准备一个栈,标记为stack, 栈中放的元素是数组的位置,开始时stack为空。

如果找到每一个i位置左边和右边离i位置最近且值比arr[i]小的位置,那么需要让stack从栈顶到栈底的位置所代表的值是严格递减的;

如果找到每一个位置左边和右边离i位置最近且值比arr[i]大的位置,那么需要让stack从栈顶到栈底的位置所代表的值是严格递增的

def getNearNorepeat(L):
    res = [[0,0]]*len(L)
    
    stack = []

    for i in range(len(L)):
        while len(stack)!=0 and L[stack[-1]] >  L[i]:
            popIndex = stack.pop()

            if len(stack) == 0:
                lessLeftIndex = -1
            else:
                lessLeftIndex = i

            res[popIndex][0] = lessLeftIndex
            res[popIndex][1] = i

        stack.push(i)
    
    while len(stack)!=0:
        popIndex = stack.pop()
        if len(stack) == 0:
            lessLeftIndex = -1
        else:
            lessLeftIndex = stack[-1]

        res[popIndex][0] = lessLeftIndex
        res[popIndex][1] = -1

    return res
def getNearLess(L):
    res = [[0,0]] * len(L)
    stack = []
    
    for i in range(len(L)):
        while len(stack)!=0 and L[stack[-1][0]] > L[i]:
            popIndex = stack.pop()
            
            if len(stack)==0:
                leftLessIndex = -1
            else:
                leftLessIndex = stack[-1][len(stack[-1])-1]
                
            for index in popIndex:
                res[index][0] = leftLessIndex
                res[index][1] = i
                
        if len(stack)!=0 and L[stack[-1][0]] == L[i]:
                stack[-1].append(i)
                
        else:
            list_ = []
            list_.append(i)
            stack.push(list_)
            
    while len(stack)!=0:
        popIndex = stack.pop()
        if len(stack) == 0:
            leftLessIndex = -1
        else:
            leftLessIndex = stack[-1][len(stack[-1])-1]
        
        for index in popIndex:
            res[index][0] = leftLessIndex
            res[index][1] = -1
            
    return res

整个过程中,进栈一次,出栈一次,所以整个流程时间复杂度为O(N)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
单调栈是一种特殊的栈结构,它的主要应用是用于解决求某个元素的左边右边一个比它大或小的元素的问题。单调栈的特点是栈中的元素保持有序,即栈顶元素最小或最大。而发射站是一个具体的应用场景,我们可以利用单调栈来解决相关问题。 在一个发射站中,我们需要找到每个位置的第一个比它大的位置。我们可以使用单调栈来实现这个功能。首先,我们将第一个元素的下标入栈。然后,依次遍历剩下的每个元素。对于每个元素,我们将它与栈顶元素进行比较。如果当前元素比栈顶元素大,说明找到了以栈顶元素为高度的发射站,我们可以更新它对应的输出结果,并弹出栈顶元素。直到当前元素小于或等于栈顶元素,我们将当前元素的下标入栈。这样,最后栈中剩下的元素对应的输出结果为-1,表示没有找到比它大的位置。 使用单调栈来解决发射站问题的时间复杂度为O(n),其中n为发射站的个数。这是因为每个元素最多入栈一次,出栈一次,所以遍历所有元素的时间复杂度为O(n)。而每个元素入栈和出栈的操作时间复杂度为O(1)。因此,总的时间复杂度为O(n)。 综上所述,利用单调栈结构可以高效地解决发射站问题。通过遍历发射站中的元素,可以找到每个位置的第一个比它大的位置。这种方法的时间复杂度为O(n),其中n为发射站的个数。因此,使用单调栈一个有效的解决方案。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值