leetcode链接https://leetcode-cn.com/problems/trapping-rain-water/
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
如下图,本题重要的不是动态规划的思想,而是几何思维解题的方法
- step1.从左向右遍历找到最大
- step2.从右向左遍历找到最大
- step3.加起来,并删去多余的部分
定义:dp_left[i]表示从左向右当前最大值,dp_right[i]表示从右向左当前最大值
边界:可不定义边界,初始化为dp_left[0]=0,dp_right[N-1]=0
状态转移方程:dp_left[i] = max(dp_left[i-1],nums[i]) dp_right[i] = max(dp_right[i+1],nums[i])
class Solution:
def trap(self, height: List[int]) -> int:
# 边界条件
if not height: return 0
n = len(height)
maxleft = [0] * n
maxright = [0] * n
ans = 0
# 初始化
maxleft[0] = height[0]
maxright[n-1] = height[n-1]
# 设置备忘录,分别存储左边和右边最高的柱子高度
for i in range(1,n):
maxleft[i] = max(height[i],maxleft[i-1])
for j in range(n-2,-1,-1):
maxright[j] = max(height[j],maxright[j+1])
# 一趟遍历,比较每个位置可以存储多少水
for i in range(n):
if min(maxleft[i],maxright[i]) > height[i]:
ans += min(maxleft[i],maxright[i]) - height[i]
return ans
作者:z1m
链接:https://leetcode-cn.com/problems/trapping-rain-water/solution/dong-tai-gui-hua-shuang-zhi-zhen-tu-jie-by-ml-zimi/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
代码优化(一次遍历,优化空间,思路跟最大子序列问题一致):
class Solution:
def trap(self, height: List[int]) -> int:
Left_h,Right_h = 0 ,0
N = len(height)
ans = 0
for i in range(N):
Left_h = max(Left_h, height[i])
Right_h= max(Right_h,height[N-i-1])
ans += Left_h + Right_h - height[i]
return ans - N*Left_h