[leetcode] 42. 接雨水

题目描述

给定 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⁴
• 0 <= height[i] <= 10⁵

解题方法

方法一：栈

看完题目，首先需要想到的是什么情况下会产生积水？

那我们可以思考一种场景。假设有三根柱子，从左往右数第一根柱子高度大于第二根柱子，第三根柱子也大于第二根柱子，此时第一根柱子与第三根柱子之间是不是就有积水了。

而根据第三根柱子的高度，又分为以下两种情况讨论。

如上图所示，当第三根柱子高度大于第一根柱子时，此时不管第三根柱子后面还有多少柱子，第一根柱子与第三根后面的柱子之间都不会再产生积水了。

如上图所示，当第三根柱子高度小于第一根柱子时，第一根柱子还可能与第三根后面的柱子之间产生积水，此时满足的条件是第三根柱子后面必须要有比自己高的柱子。

当我们看到以上两种情况，我们是不是就可以列出其中的规律了。

我们可以用栈保存每根柱子。

当栈为空时，我们将柱子入栈，当后面的柱子比前面入栈的柱子矮时，我们持续将柱子入栈。

而当后面的柱子比栈顶的柱子高时，我们将栈顶的柱子出栈，栈顶的柱子出栈后，我们需要计算此时栈顶的柱子与当前柱子之间形成的积水容量。怎么计算呢？我们可以知道两根柱子之间的距离，{两根柱子的距离} * {两根柱子较低柱子的高度 - 出栈柱子的高度}，即为形成积水的容量。

柱子出栈后，若当前柱子还是比栈顶柱子高时，我们重复上面的操作，计算两根柱子之间的积水容量。直到栈为空或者当前柱子比栈顶柱子矮时，则将当前柱子加入栈中。

然后继续往后遍历，重复上面的步骤，最终计算出所有积水容量。

思路我说完了，下面看下代码吧。

java代码

public int trap(int[] height) {
    int area = 0;
    Stack<Integer> stack = new Stack<>();
    for(int i = 0; i<height.length; i++) {
        // 当前柱子比栈顶柱子高时，栈顶柱子出栈，计算积水容量。
        while (!stack.isEmpty() && height[i] > height[stack.peek()]) {
            int lastIndex = stack.peek();
            stack.pop();
            if(stack.isEmpty()) {
                break;
            }
            area = area + (Math.min(height[i], height[stack.peek()]) - height[lastIndex]) * (i - stack.peek() - 1);
        }
        // 栈为空时，当前柱子入栈
        if (stack.isEmpty()) {
            stack.push(i);
        } else {
            // 当前柱子比栈顶柱子矮，当前柱子入栈
            if (height[i] < height[stack.peek()]) {
                stack.push(i);
            }
            // 当前柱子跟栈顶柱子一样高，栈顶柱子出栈，新柱子入栈（为了计算积水容积时能够套用上面的计算公式）
            else {
                stack.pop();
                stack.push(i);
            }
        }
    }
    return area;
}

复杂度分析

时间复杂度：$O(n)$，循环遍历一次，栈中元素最多出栈入栈一次。
空间复杂度：$O(n)$，存储结构用到了栈，最坏情况下会将所有元素入栈。

---

• 个人公众号
  
• 个人小游戏