Trapping Rain Water

最新推荐文章于 2024-09-17 16:23:14 发布

yangmicheal

最新推荐文章于 2024-09-17 16:23:14 发布

阅读量262

点赞数

CC 4.0 BY-SA版权

分类专栏： leetCode Solutions

本文链接：https://blog.youkuaiyun.com/yangmicheal/article/details/40072637

leetCode Solutions 专栏收录该内容

53 篇文章

订阅专栏

本文介绍了一种用于计算二维高度图中能捕获多少水量的算法。通过预计算每个位置左侧和右侧的最大高度，该算法有效地确定了每个位置可以捕获的水量，并最终返回总的捕获水量。

public class Solution {
    public int trap(int[] A) {
        int[] maxL = new int[A.length];
        int[] maxR = new int[A.length];
        int max = 0;
        // Go left and find left max height for current point
        for (int i=0; i<A.length; i++){
            maxL[i] = max;
            if (max<A[i]){
                max=A[i];
            }
        }
        max = 0;
        // Go right and find right max height for current point
        for (int i=A.length-1; i>=0; i--){
            maxR[i] = max;
            if (max<A[i]){
                max = A[i];
            }
        }
        
        int capacity = 0;
        // Iterate through A, capacity of current point = min (maxLeftHeight, maxRightHeight) - currentHeight
        for (int i=0; i<A.length; i++){
            int c = Math.min(maxL[i], maxR[i]) - A[i];
            if (c>0) {
                capacity+=c;
            }
        }
        return capacity;
    }
}