Java实现 LeetCode 42 接雨水

42. 接雨水

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

class Solution {
    public int trap(int[] height) {
         int n = height.length, sum = 0;
        if(n == 0) return 0;
        int[] left = new int[n];
        int[] right = new int[n];
        left[0] = height[0];
        right[n-1] = height[n-1];
        /*思路：动态规划求解
         * 计算出每一列中左边最高， 右边最高的高度
         * 最后每一列中包含水滴的个数即为两边最低的高度减去当前高度
        */
        for(int i = 1; i < n; i ++) {
        	left[i] = Math.max(height[i], left[i-1]);
        }
        for(int i = n-2; i >= 0; i --) {
        	right[i] = Math.max(height[i], right[i+1]);
        }
        //计算每一列中水滴的数量
        for(int i = 0; i < n; i ++) {
        	sum += Math.min(left[i], right[i])-height[i];
        //	System.out.println(Math.min(left[i], right[i])-height[i]);
        }
        return sum;
    }
}