Trapping Rain Water

最新推荐文章于 2022-08-22 20:23:40 发布

weixin_34117522

最新推荐文章于 2022-08-22 20:23:40 发布

阅读量44

点赞数

本文介绍了一种计算雨后地面积水总量的高效算法。针对由非负整数构成的高度图，通过预先计算左右两侧的最大高度，实现了快速求解雨水能够积存的总量。文章详细解释了算法的实现思路，并提供了完整的Java代码示例。

题目

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

方法

对于每个A[i]，trap water 为 min（left， right）- A[i];

    public int trap(int[] A) {
        if (A == null || A.length == 0 || A.length == 1 || A.length == 2) {
            return 0;
        }
        int len = A.length;
        int[] left = new int[len];
        int[] right = new int[len];
        int leftMax = 0;
        for (int i = 0; i < len; i++) {
        	left[i] = leftMax;
        	if (A[i] > leftMax) {
        		leftMax = A[i];
        	}
        }
        int rightMax = 0;
        for (int i = len; i > 0; i--) {
        	right[i - 1] = rightMax;
        	if (A[i - 1] > rightMax) {
        		rightMax = A[i - 1];
        	}
        }
        
        int sum = 0;
        for (int i = 0; i < len; i++) {
        	int temp = Math.min(left[i], right[i]) - A[i];
        	if (temp > 0) {
        		sum += temp;
        	}
        }
        return sum;
    }