力扣接雨水（解析）

萧木易2023-09-14 10:35

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：

复制代码

输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

示例 2：

复制代码

输入：height = [4,2,0,3,2,5]
输出：9

找到一个高点，轮训到下一个高点，记录高点的值和下标

相邻的两个高点比较，取最小值

两个高点下标的差值

相邻两个高点的内值总和

两个高点的容量等于最小值乘以距离差

两个高点的接雨水等于容量减去两个高点内值总和

轮训加上所有雨水值

hight = $0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1 ,3$

#记录高点值

hig_dict ={}

for i in range(len(hight)):

if i+1==len(hight):

如果最后一个值大于前一个值

if hight $i$ >hight $i-1$ ：

hig_dict $i$ =hight $i$

if higth $i$ >hight $i+1$ and higth $i$ >hight $i-1$ :

hig_dict $i$ =hight $i$

else:

pass

#高点值下标

keys = list $hig_dict.keys$

#高点值

values = list $hig_dict.values$

#所有高点值容量

sum_all =\[\]

for i in range(len(values)):

if i+1==len(values):

break

#取相邻两个高点的最小值

if values $i$ <=values $i+1$

tmp = values $i$

else：

tmp = values $i+1$

#-1是因为要去相邻两个数之间有几个值

length = keys $i+1$ - keys $i$ -1

sum_all.append(tmp*length)

#相邻高点内包含值的总容量

sum_out =\[\]

for k,v in enumerate(hight):

#排除最小高点之前的值

if k < keys $0$ :

continue

#排除最大高点之后的值

if k > keys $-1$ :

break

#排除高点值

if k in keys:

continue

else:

sum_out.append(hight $k$ )

sum_capacity = sum(sum_all) - sum(sum_out)