【LeetCode】直方图水量

1、题目链接

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 1。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的直方图，在这种情况下，可以接 6 个单位的水（蓝色部分表示水）。

原题地址。

2、思路

我把这个题目视为递归寻找“子池”的问题。所谓子池，指的是能够一个能够单独储水、且与其他子池的水没有交集的区域。

在上面的图中，一共可以找到三个子池，如下图所示：

可以想见，如果找到了子池的边界（即索引）与子池两侧“墙”的高度（即该出的值），那么求解子池的储水量是十分简单的。于是，重点就是如何找到这些子池。

要想子池能够储水，我们首先要找到子池的“左墙”。不难想到，左墙的特点为：

左墙应该比其右侧相邻处的值大

找到左墙之后，接下来就是找到对应于这个左墙的右墙（如果存在的话），使之构成一个能够储水的子池。右墙的特点略微复杂一些：

• 右墙的值可以大于等于左墙（如上图中第一个子池和第二个子池）
• 右墙的值可以小于左墙，但只需大于这段区间内的某个值，仍可以构成一个新的子池

第一个特点很好理解，关键是第二个。

拿上图中的第三个子池举例：按照寻找左墙的方法，会把位置7视为左墙，因为它比位置8要高，但是从位置7开始向右寻找，不能找到等于或超过7的墙作为右墙，这时，需要降低7处的高度一个单位，再向右找右墙。在这个例子中，降低7处的墙一个单位后，发现其与8处的值无法满足构成左墙的条件了，这时，可以跳过7，从8开始执行新的查找算法了。

之所以需要降低左墙一个单位的高度，是为了考虑出现如[4,2,3]的情况。实际上，这个数组也可以构成一个子池。

3、代码

python代码如下（没有进行进一步的优化）：

class Solution:
    def trap(self, height: List[int]) -> int:
        i = 0  # 初始化循环的向量
        res = 0  # 记录储水量的值
        while i < len(height) - 1:
            if height[i + 1] >= height[i]:  # 不满足作为左墙的条件
                i += 1
                continue
            else:
                left_val = height[i]  # 记录左墙的值
                left_ind = i          # 记录左墙的位置
                right_val, right_ind = None, None
                for j in range(i + 2, len(height)):  # 遍历是否可以找到右墙
                    if height[j] >= left_val:
                        right_val = height[j]
                        right_ind = j
                        break
                if right_val is not None:  # 如果找到了右墙，那么一切简单
                    max_val = min(left_val, right_val)
                    for k in range(left_ind+1, right_ind):
                        res += max_val - height[k]
                    i = right_ind
                else:  # 没有找到与左侧边界同样高的墙，则对左墙高度减一继续查找
                    left_val -= 1
                    flag = True
                    while left_val > height[left_ind + 1]:  # 只要仍是"墙"
                        for j in range(i+2, len(height)):
                            if height[j] >= left_val:
                                right_val = height[j]
                                right_ind = j
                                flag = False
                                break
                        if not flag:
                            break
                        else:
                            left_val -= 1
                    if right_val is not None:  # 减一后找到了对的得右墙
                        max_val = min(left_val, right_val)
                        for k in range(left_ind+1, right_ind):
                            res += max_val - height[k]
                        i = right_ind
                    else:  # 仍没有找到，则说明当前左墙的位置不可能再有与之构成子池的右墙了
                        i += 1
        return res