题意:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
思路:
我们把每个下标对应的位置当成独立的个体来看,对于当前位置能接水的体积,根据短板原理,是由左右两边最矮的那个来决定的,因此对于每个位置能接水的体积,是他左边最高的柱子与右边最高的柱子两者取小。但是两层循环是超时的,我们可以预处理,提前处理出左右两边到当前位置的最大值。
public int trap(int[] height) {
int maxn=30005;
int maxl[]=new int[maxn],maxr[]=new int[maxn];
Arrays.fill(maxl,0);
Arrays.fill(maxr,0);
for(int i=1;i<height.length;i++){
maxl[i]=Math.max(maxl[i-1],height[i-1]);
}
for(int i=height.length-2;i>=0;i--){
maxr[i]=Math.max(maxr[i+1],height[i+1]);
}
int ans=0;
for(int i=0;i<height.length;i++){
ans+=Math.max(Math.min(maxl[i],maxr[i])-height[i],0);
}
return ans;
}
时间复杂度O(N)
空间复杂度O(N)