#42:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water/
这题属于困难,参考了解题区一位数学大佬的解题思路:https://leetcode-cn.com/problems/trapping-rain-water/solution/wei-en-tu-jie-fa-zui-jian-dan-yi-dong-10xing-jie-j/,本来很困不想写了,看完这个思路以后就觉得简单很多,果然人脑和人脑是不一样的。整理一下,大概如下:
解题思路:
1. 简单举个例子:height=[2,1,0,3,1],如下图:
2. 下雨方向从左往右,依次填满,S1=墙的面积+雨水的面积=2+2+2+3+3 =12
下雨方向从右往左,依次填满,S2=墙的面积+雨水的面积=1+3+3+3+3 =13
3. 这部分可以根据**韦恩图(venn)**来理解,如下图,S1+S2 包含了整个矩形,而且重合部分正好是墙和水的面积加总,所以:S1 + S2 = S墙 + S水 + S矩形,所以:S水 = S1 + S2 - S矩形 - S墙
4. S矩形 = list的长度 × list内最大值 = 5×3 = 15,S墙 = list内数字的加总 = sum[2,1,0,3,1] = 7,S水 = 12 + 13 - 15 - 7 = 3,和上面蓝色的3格相符,应该是正确的,最后写code测测看是不是这样就可以了。
实测:
class Solution(object):
def trap(self, height):
"""
:type height: List[int]
:rtype: int
"""
#方向从左往右
s1=0 #初始面积设为0
l1=0 #方向从左往右,墙的最大高度,初始设为0
for i in range(len(height)):#i分别是第0,1,2,3,4根墙柱子
if height[i]>l1:#如果第i根墙柱子大于目前最大高度l1,
l1=height[i]#则最大高度变为第i根墙柱子
#s1+=l1
s1+=l1#加总
#print(s1)
#方向从右往左,原理和上面一样
height2=height[::-1]#list倒序排列, 模拟从右往左的墙,height=[2,1,0,3,1],所以height2=[1,3,0,1,2]
s2=0
l2=0
for j in range(len(height2)):
if height2[j]>l2:
l2=height2[j]
#s2+=l2
s2+=l2
#print(s2)
#print(s1+s2-len(height)*max(height)-sum(height))
return s1+s2-len(height)*max(height)-sum(height)
leecode实测截图: