给定 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)