python 维护单调栈来达到某些目的

在维护一个单调增的栈时，可以找到数组array中右边比他大的第一个元素，那么如果想要找到数组array 中右边比他小的元素，则可以通过维护一个单调减的栈来实现目的，代码对数组的遍历均为从前往后遍历

# 例如：array = [3,2,4,1] 结果ans返回一个数组，数组为对应位置右边第一个比他小的数的下标，
# 没有找到则对应位置返回-1
# ans = [1,3,3,-1]

array = [3,2,4,1] 
n = len(array)
ans = [-1] * n
stack = []
for i in range(n):
    num = array[i]
    while stack and array[stack[-1]] > num:
        x = stack.pop()
        ans[x] = i # i指向的是比数组中下标x右边第一个比其小的数的下标
    stack.append(i)
print(ans) # [1,3,3,-1]

有时候，我们还希望找到遍历到的当前数左边的第一个比他小的数从而达到解决实际问题的目的，那么怎么得到比当前遍历到的数左边第一个比其小的数呢，下面给出代码和解释

 

array = [2,3,1,2] # [-1,0,-1,2]
ori_n = len(array)
array.append(0)   # 在数组的开始加入0，否则当数组单调递增时无法跳出while循环进行操作
array.insert(0,0) # 在数组的末尾加入0  否则当数组单调递减是会出问题
n = len(array)
ans = [-1] * n
stack = []
for i in range(n):
    num = array[i]
    while stack and array[stack[-1]] > num:
        x = stack.pop() # 遍历到的元素左边的元素下标
        if len(stack) != 0:  # 判断栈不为空
            ans[x] = stack[-1]
        else:
            continue
    stack.append(i)
ans = [i - 1 for i in ans][1:ori_n + 1]

print(ans) 

若是将数组反转，或者说逆着遍历，也可以找到右边比当前遍历到的元素小的第一个数，此时维护的是一个单调递增的栈，代码如下

array = [3,1,4,2] # 找到对应数字右边比他小的第一个数的下标，若找不到，在对应位置返回-1
n = len(array)
ans = [-1] * n
stack = []  # 此时维护的是一个单调递增的栈
for i in range(n-1,-1.-1):
    num = array[i]
    while stack and num[stack[-1]] < num:
        x = stack.pop()
        ans[i] = x
    stack.append(i)
print(ans) # [1,-1,3,-1]