给定一个直方图(也称柱状图),假设有人从上面源源不断地倒水,最后直方图能存多少水量?直方图的宽度为 1。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的直方图,在这种情况下,可以接 6 个单位的水(蓝色部分表示水)。 感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
题解
class Solution:
def trap(self, height: List[int]) -> int:
length = len(height)
contain = 0
for i in range(length):
if i == 0:
next
elif i == length-1:
next
else:
left_max = max(height[:i])
right_max = max(height[i:])
contain_now = 0 if height[i]>= min(left_max,right_max) else min(left_max,right_max)-height[i]
contain = contain + contain_now
return contain