leetcode 42. 接雨水

42. 接雨水

尝试了了一下 大名鼎鼎的接雨水。

分析一下，是不是能接雨水的条件在于是否存在一个低洼处，也就是一个height数组的变化拐点，所以按照正常思路去想就是找这个拐点。能够存储水的容量和这个低洼处的左边与右边的值有关系，所以如果这么找的话，就是按照列来算，那么需要向左右两边去扫最高点，复杂度在O(n2)。

如果换个思路，按照行来扫的话，就不用去扫右边的最高点，只需要去找左面的拐点就行了。那就是用一个栈维护向下递减的高度，但新出现的高度违反了单调性，那么就可以把之前储存进来的值拿出来进行计算。用这样的算法，叫做单调栈。

代码如下：

package pro42;

import java.util.Stack;

public class Solution {

    public int trap(int[] height) {
        Stack<Integer> stack = new Stack<>();
        int ans = 0;
        for(int i = 0; i < height.length; i++) {
            if(stack.isEmpty()) {
                stack.add(i);
                continue;
            }
            if(height[i] <= height[stack.peek()]) {
                stack.add(i);
            } else {
                while(!stack.isEmpty() && height[stack.peek()] < height[i]) {
                    if(stack.size() == 1) {
                        stack.pop();
                        break;
                    }
                    int cbottom = stack.pop();
                    int left = stack.peek();
                    ans += (Integer.min(height[left], height[i]) - height[cbottom]) * (i - left - 1);
                }
                stack.add(i);
            }
        }
        return ans;
    }

    public static void main(String[] argv) {
        System.out.println(new Solution().trap(nums));
    }

//    public static int[] nums = new int[]{0,1,0,2,1,0,1,3,2,1,2,1};

    public static int[] nums = new int[]{4,2,0,3,2,5};
}

同样的题还有 84. 柱状图中最大的矩形