Leetcode 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

这道题立马让我想到了之前做过的一道求两条柱子间面积的题。此题使用了类似的方法。我们不用仔细地计算每个"容器的装水"容量、我们只需要计算装水前的容器面积和装满水后的容器面积之差即可。此题的难点是如何装满水成为一个新nums。在此我使用了从头和尾夹逼的方法，从低往高装水。装完一层形成新的数组再装下一层~

public class Solution {
    public int trap(int[] height) {
        int preS=0;    //原面积
        int aftS=0;    //后面积
        int pre=0;
        int aft=height.length-1;
        int temp;
        for(int i=0;i<height.length;i++){
            preS+=height[i];
        }
        while(pre<aft){
            if(height[pre]<height[aft]){
                temp=height[pre];
            }else{
                temp=height[aft];
            }
            for(int j=pre+1;j<aft;j++){
                height[j]=Math.max(temp, height[j]);
            }
            if(height[pre]<height[aft]){
                while(pre<aft&&height[pre]<=temp){
                    pre++;
                }
            }else{
                while(aft>pre&&height[aft]<=temp){
                    aft--;
                }
            }
        }
        for(int i=0;i<height.length;i++){
            aftS+=height[i];
        }
        return aftS-preS;
    }
}