题目内容
Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。
Example:
Input: [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
题目思路
认真观察所给的蓄水图,可以发现水的高度依赖于木桶效应。也就是和两侧中较低的那一侧有关系。所以我们可以可以找到其中最高的一根作为最高侧。然后两侧分别向中间遍历。对于左侧和右侧而言,如果当前的木板高度小于之前所遍历的最高高度,那么就做差来计算水量,否则该点成为新的最大高度,直到遇到最高位置。。之所以不用中间到两侧遍历是因为要从远离最高柱子的向内才能衡量高度。
程序代码
class Solution(object):
def trap(self, height):
"""
:type height: List[int]
:rtype: int
"""
#找到最高的那个
#两侧向最高开始收拢
#两侧面积相加即可
if len(height)<=2:
return 0
highest=max(height)
index=height.index(highest)
left,right=0,len(height)-1
larea,rarea=0,0
ltemp,rtemp=height[left],height[right]
while left<index or right>index:
if left<index:
if height[left]<ltemp:
larea+=-height[left]+ltemp
#print(larea)
else:
ltemp=height[left]
left+=1
if right>index:
if height[right]<rtemp:
rarea+=-height[right]+rtemp
#print(rarea)
else:
rtemp=height[right]
right-=1
#print(larea,rarea)
return larea+rarea