0.题目
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5]
输出:9
提示:
n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105
1.思路
首先,显然易知,第一个与最后一个区域无法接收任何雨水,所以,可以接受雨水的范围是[1:n-2]。
我们可以观察出,某一区域可接收雨水的量取决于它左边最高的柱子与右边最高的柱子,若min(左边最高的柱子高度,右边最高的柱子高度)>该区域柱子高度,则可以接到雨水,否则无法接到雨水。
我们可以直接暴力求解,对于每个区域,找到它左边最高的柱子和右边最高的柱子。该方法时间复杂度为O(n^2),空间复杂度为O(1)。经过尝试,该方法TLE
仔细思考,我们可以开两个长度为n的数组,分别存储各个区域左边最高的柱子和右边最高的柱子的高度,可以利用dp求解
2.代码
class Solution:
def trap(self, height: List[int]) -> int:
if not height: #python中的空列表与数组均为NULL
return 0
n=len(height)
leftmax=[0]*n
for i in range(1,n-1):
leftmax[i]=max(leftmax[i-1],height[i-1])
rightmax=[0]*n
for i in range(n-2,0,-1):
rightmax[i]=max(rightmax[i+1],height[i+1])
return sum((max(0,min(leftmax[i],rightmax[i])-height[i])) for i in range(1,n-1))