接雨水-Java实现

问题描述

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

思路解析

两边取最大值，遍历 空间O(n) 时间 O(n)

class Solution {
    public int trap(int[] height) {
        if (null == height || height.length == 0) {
            return 0;
        }

        int[] l2R = new int[height.length];
        int[] r2L = new int[height.length];

        int max = 0;
        for (int i = 0; i < height.length; i++) {
            max = height[i] > max ? height[i] : max;
            l2R[i] = max;
        }

        max = 0;
        for (int i = height.length - 1; i >= 0; i--) {
            max = height[i] > max ? height[i] : max;
            r2L[i] = max;
        }

        int result = 0;
        for(int i = 0; i < height.length; i++){
            result += Math.min(l2R[i], r2L[i]) - height[i];
        }

        return result;
    }
}