LeetCode42. 接雨水

题目

给定 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

分析

1.自己一开始的想法：把上图分为一层一层，每一层的高度为1,用栈分别计算每层的凹进去格子个数，加在一起，哈哈哈哈哈，数据量小还可以用，数据量一大，就超时了，时间复杂度O(n^2)。

2.还是看了解题报告，只要遍历三次，就可以得出结果，哇塞好神奇。

其实这个题只要能知道第x个柱子的 左边最高柱子 和 右边最高柱子 中比较矮的一个，然后用较矮柱子的高度减去当前第x个柱子的高度，就是当前x柱子可以接住的雨水量。

说的有点儿绕，写成表达式就是： x柱子接住雨水量 =  min( left_max, right_max) - height[x]

代码

class Solution {
    public int trap(int[] height) {
        int len = height.length;
        if (len == 0) return 0;
        int[] left_max = new int[len];
        int[] right_max = new int[len];

        int max = height[0];
        for (int i = 0; i < len; i++) {
            if (max > height[i]){
                left_max[i] = max;
            }else{
                left_max[i] = height[i];
                max = height[i];
            }
        }

        max = height[len-1];
        for (int i = len-2; i >= 0; i--) {
            if (max > height[i]){
                right_max[i] = max;
            }else{
                right_max[i] = height[i];
                max = height[i];
            }
        }

        int sum = 0;
        for (int i = 1; i < len-1; i++) {
            sum += Math.min(left_max[i],right_max[i]) - height[i];
        }

        return sum;
    }
}