python 栈的应用---接雨水

目录

   题目

   栈：

   思路：

   结论

   代码

---

题目

在力扣上有一道算法题叫做接雨水。如下图（图片来自于力扣），要计算出在柱子的缝隙中能存多少水。

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

栈：

一种运算受限的线性表，遵循后进先出的规则。如下图（图片来自于百度百科）

思路：

计算列表list=[5,3,1,5]能接多少雨水。

1.创建一个stack=[]

2.从左向右遍历列表list，将list元素的index，0插入stack中，如果list[1]<=stack[-1]的元素，则继续将1插入stack。继续遍历插入，直到得到list[i]>=stack[-1]，然后执行stack.pop()。此时开始计算在list中，下标为stack[-1]与下标为i之间的雨水数。值得注意的是此时的stack[-1]是执行完stack.pop()之后的。举个例子s=[1,2,3]，s[-1]=3，如果执行完一次s.pop()后，s[-1]=2。可以先了解一下pop的用法。

3.然后继续比较list[stack[-1]]与list[i]的的高度，如果list[stack[-1]]>list[i]，则stack.pop()，并计算此时stack[-1]与i之间的雨水高度。

（对于list=[5,3,1,5]这个列表第2步计算得到图中标号为1的雨水面积，第3步计算得到图中面积为2的雨水面积，然后汇总得到结果）

4.直到遍历结束

结论

解法中的关键在于stack的弹出，不断循环比较list[i]与栈中最后一个元素的大小，如果栈中最后一个元素小于list[i]，则说明在list[i]和stack[-2]之间有可能存水，因为stack[-2]>=stack[-1]

代码

class Solution:
    def trap(self, height) -> int:
        s=0
        if len(height)<3: return s
        stack=[]
        for i in range(len(height)):
            while len(stack)> 0 and height[i] > height[stack[-1]]:
                p=stack.pop()
                if len(stack)==0:
                    break
                s+=(min(height[stack[-1]],height[i])-height[p])*(i-stack[-1]-1)
            stack.append(i)
        return s