LeetCodeDifficult-【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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路一：暴力：直接扫描每一个高度，看是否有封闭区间，封闭区间则加上区间长度。

   提交超时!!!

class Solution:
    def trap(self, height: List[int]) -> int:
        if height == []:
            return 0
        ans = 0
        for h in range(1, max(height)+1):
            op = -1
            for j in range(len(height)):
                if op == -1 and height[j] >= h:
                    op = j
                    # print(op, j, ans)
                elif op != -1 and height[j] >= h:
                    ans += j - op - 1
                    # print(op, j, ans)
                    op = j
        return ans

官方题解中的暴力,也超时了：

class Solution:
    def trap(self, height: List[int]) -> int:
        if height == []:
            return 0
        ans = 0
        size = len(height)
        for i in range(size):
            left = 0
            right = 0
            for j in range(i, -1, -1):
                left = max(left, height[j])
            for j in range(i+1, size):
                right = max(right, height[j]) 
            if min(left, right) - height[i] > 0: 
               ans += min(left, right) - height[i]
        return ans

思路二：

方法 2：动态编程

在暴力方法中，我们仅仅为了找到最大值每次都要向左和向右扫描一次。但是我们可以提前存储这个值。因此，可以通过动态编程解决。

class Solution:
    def trap(self, height: List[int]) -> int:
        if height == []:
            return 0
        ans = 0
        size = len(height)
        
        max_left = [0] * size
        max_right = [0] * size
        for i in range(1, size):
            max_left[i] = max(max_left[i-1], height[i-1])
        for i in range(-2, -size, -1):
            max_right[i] = max(max_right[i+1], height[i+1])
        for i in range(size):
            if min(max_left[i], max_right[i]) - height[i] > 0: 
               ans += min(max_left[i], max_right[i]) - height[i]
        return ans

方法 3：使用双指针

方法 4：栈的应用