1.接雨水问题
给定 n 个非负整数表示每个宽度为 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):
"""
:type height: List[int]
:rtype: int
"""
second = 0
result = 0
maxid = 0
for i in range(1, len(height)):
if height[i] > height[maxid]:
maxid = i
for i in range(0, maxid):
if second < height[i]:
second = height[i]
else:
result += second - height[i]
second = 0
for j in range(len(height) - 1, maxid, -1):
if second < height[j]:
second = height[j]
else:
result += second - height[j]
return result