LeetCode42接雨水问题

LeetCode42

首先，单独来看每一个柱子，每一个柱子上边能乘的雨水和三个东西有关系。左边的最大值和右边的最大值和自己的高度。用公式来表示就是：

curWater = Min(lMax, rMax) - curHeight

思路清晰了之后，这道题有很多解法。比如比较麻烦的方法先遍历一遍找出最大值，之后在最大值左边和右边遍历。最大值左边的rMax就是这个最大值。这种方法需要遍历两遍。不是最优解。

还有一种方法是可以遍历一遍就出结果的。那就是单调栈。

单调栈就是维护一个“头重脚轻“的栈。如果有大于栈顶的元素进入则说明可以乘住雨水了。把小于当前元素的都pop出来。

栈中存储的是数组元素的index。代码如下：

public class Solution {
    public int trap(int[] height) {
        int res = 0;
        if (height == null) {
            return 0;
        }
        int len = height.length;
        // 单调栈
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < len; i++) {
            // 如果不单调的话那就持续pop
            while (!stack.empty() && height[stack.peek()] < height[i]) {
                Integer curIdx = stack.pop();
                // 如果栈顶元素一直相等，那么全都pop出去，只留第一个。
                while (!stack.isEmpty() && height[stack.peek()] == height[curIdx]) {
                    stack.pop();
                }
                // 获取栈数据的时候总要看是否为空
                if (!stack.empty()) {
                    Integer left = stack.peek();
                    res += (Math.min(height[left], height[i]) - height[curIdx]) * (i - left - 1);
                }
            }
            // 单调的话就继续push
            stack.push(i);
        }
        return res;
    }
}