给定 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 * 10^4
0 <= height[i] <= 10^5
方法:双指针
class Solution:
def trap(self, height: List[int]) -> int:
#思路:遍历height数组,找到每个点的左边最大和右边最大,然后取他们俩的小的一方作为高
n=len(height)
if n<3:
return 0
ans=0
left_max = [0] * n # 用于存储每个点的左边最大高度
right_max = [0] * n # 用于存储每个点的右边最大高度
# 计算每个点的左边最大高度
max_left = 0
for i in range(n):
left_max[i] = max_left
max_left = max(max_left, height[i])
# 计算每个点的右边最大高度
max_right = 0
for i in range(n - 1, -1, -1):
right_max[i] = max_right
max_right = max(max_right, height[i])
# 计算每个点上能存储的水量
for i in range(1, n - 1):
min_side = min(left_max[i], right_max[i])
if min_side > height[i]:
ans += min_side - height[i]
return ans