题目
给定 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来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
1. 原始解法
本体的主要思路就是,每一格可以容纳的雨水,应当是它左边最高的边界和右边最高的边界之间较小的一个,减去它本身高度的差值。比如以示例1中的第5格为例,它左边最高的边界是2,右侧最高的边界是3,下雨后,最多容纳高度为2的雨水。同时它本身的高度是0,所以容纳了两格雨水。
基于这个思想,我们使用两个数组记录每一格左右两侧最高的边界。
left = [-1 for i in range(length)]
right = [-1 for i in range(length)]
首先从左侧开始遍历数组,记录下最高的边界。
lm = -1 # left max
for i in range(1, length):
if height[i - 1] > lm:
lm = height[i - 1]
left[i] = lm
右侧的操作类似。最后再循环一次,计算每格容纳的雨水,求和即可。注意,对于本身高度比左右两侧都高的格子,它容纳的雨水应当是0,而非负值。
for i in range(1, length-1):
height_i = min((left[i], right[i]))
answer += max((height_i - height[i]), 0)
结果:通过测试用例,用时52ms,内存使用量15.8MB
完整代码:
class Solution:
def trap(self, height: List[int]) -> int:
length = len(height)
left = [-1 for i in range(length)]
right = [-1 for i in range(length)]
lm = -1 # left max
rm = -1 # right max
for i in range(1, length):
if height[i - 1] > lm:
lm = height[i - 1]
left[i] = lm
for i in range(length-2, -1, -1):
if height[i + 1] > rm:
rm = height[i + 1]
right[i] = rm
answer = 0
for i in range(1, length-1):
height_i = min((left[i], right[i]))
answer += max((height_i - height[i]), 0)
return answer
2. 第一次优化
其实,我们可以将后两个循环合并,同时也不需要记录right数组,以节省时间以及空间(虽然整体时间复杂度仍是O(n))。
结果:通关测试用例,用时32ms,内存使用量15.9MB
class Solution:
def trap(self, height: List[int]) -> int:
length = len(height)
left = [-1 for i in range(length)]
lm = -1 # left max
rm = -1 # right max
for i in range(1, length):
if height[i - 1] > lm:
lm = height[i - 1]
left[i] = lm
answer = 0
for i in range(length-2, 0, -1):
if height[i + 1] > rm:
rm = height[i + 1]
height_i = min(left[i], rm)
answer += max(height_i - height[i], 0)
return answer