浅论单调栈与单调队列-CSDN博客

本文链接：https://blog.csdn.net/m0_53437775/article/details/129567247

文章介绍了单调栈和单调队列的数据结构及其用途，它们分别用于从左至右和从右至左维护单调递减序列，并在Python中实现。单调队列用于维护大小为k的滑动窗口内的最大值和最小值。文章还提供了例题及解题代码，以及一些编程优化技巧，如初始化列表和比较运算的优化。

摘要由CSDN通过智能技术生成

一、简介

①单调栈：维持单调性，在栈顶进行元素的增删。

维护区间：只有一个端点向左移或向右移。

②单调队列：维持单调性，可在队列两端进行元素的增删。

维护区间：两个端点同时向左移或向右移，也即一个大小为k的滑动窗口。

注意：单调栈与但单调队列均可以用双端队列来维护

二、例题

单调栈

n = int(input())
data = list(map(int,input().split()))
stack = []
l = []
# 从左往右维持单调递减
for i in range(n):
    while stack and data[stack[-1]] < data[i]:
        stack.pop()
    if not stack:
        l.append(-1)
        stack.append(i)
    else:
        l.append(stack[-1] + 1)
        stack.append(i)
print(*l)
stack = []
r = []
# 从右往左维持单调递减
for i in range(n - 1,-1,-1):
    while stack and data[stack[-1]] < data[i]:
        stack.pop()
    if not stack:
        r.append(-1)
        stack.append(i)
    else:
        r.append(stack[-1] + 1)
        stack.append(i)
print(*r[::-1])

单调队列

from collections import deque
n,k = map(int,input().split())
data = list(map(int,input().split()))
g = [0] * n
g[0] = data[0]
q = deque()
q.append(0)
# 维持[max(1,i-k),i]的滑动窗口的最大值，当i > k时，就是大小为k的滑动窗口
for i in range(1,n):
    # 当窗口大于k时，就把最左端的踢出去
    if i - q[0] > k:
        q.popleft()
    # 当队列中的最右端也即最小值小于data[i],就把最右端踢出去，踢出去后如果仍小于那么接着踢出去
    # 直到把i放进队列时满足单调递减
    while q and data[q[-1]] <= data[i]:
        q.pop()
    q.append(i)
    g[i] = data[q[0]]
f = [0] * n
f[0] = data[0]
q1 = deque()
q1.append(0)
# 维持[max(1,i-k),i]的滑动窗口的最小值
for i in range(1,n):
    if i - q1[0] > k:
        q1.popleft()
    while q1 and data[q1[-1]] >= data[i]:
        q1.pop()
    q1.append(i)
    f[i] = data[q1[0]]
res = -float("inf")
for i,j in zip(g,f):
    if res < i - j:
      res = i - j
print(res)