接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

Leetcode 42.接雨水-LMLPHP

上面是由数组 [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;
}
}
05-22 04:10