LeetCode题解-接雨水
个人微信公众号:AI研习图书馆,欢迎关注~
深度学习知识及资源分享,学习交流,共同进步~
一、题目描述
42题:接雨水
难度:困难
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
二、题解实现
1. 方法一-暴力求解
1.1 解题思路
一开始拿到题目之后可能会有些手足无措,我们从最原始的暴力解法开始思考,对于每个下标 i,我们计算它上方可以接多少雨水,然后将所有下标上的雨水加起来即可。
具体来说,如图我们假设输入为 [0,1,0,2],凭感觉,我们得到接完雨水后的结果如右图。可以看到,只有下标 2 的上方可以接到雨水,其他地方的水会向两边“流掉”,所以无法接到水。
为什么是这样呢?首先看下标 1 ,它左边最高的柱子是下标 0 ,右边最高的柱子是下标 3 ,因为这两个柱子的较小值比下标 1 要低,所以下标 1 的上方没有水。相反,我们看下标 2,它左边最高的柱子是下标 1 ,右边最高的柱子是下标 3 ,这两个柱子较矮的是下标 1 ,它比下标 2 本身要高,所以下标 2 可以接到水,接的水就是高度差。
因此,我们遍历每个下标,寻找它左边和右边最高的柱子,判断是否可以接到水,将可接水的结果累加即可。
1.2 代码实现
class Solution:
def trap(self, height: List[int]) -> int:
ans = 0
for i in range(len(height)):
max_left, max_right = 0,0
# 寻找 max_left
for j in range(0,i):
max_left = max(max_left,height[j])