题目
- 接雨水
给定 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
代码(又是抄作业的一天,唉,太菜了)
class Solution {
public int trap(int[] height) {
//观察图片,其实可以发现,每个可以盛放雨水的数组“坑位”,都是前边数比后边若干数小,
//遍历数组,找到每个位置的左右边最大值,其所盛放的水量就是两边最大值的较小值-该位置高度
int len=height.length;
int ans=0;
for(int i=1;i<len-1;i++) {
int maxR=0;
int maxL=0;
for(int j=i;j>=0;j--)
maxL=Math.max(maxL,height[j]);
for(int j=i;j<len;j++)
if(height[j]>maxR)
maxR=height[j];
int a=Math.min(maxR, maxL)-height[i];
if(a>0)
ans+=a;
}
return ans;
}
}
1ms改进
//观察图片,其实可以发现,每个可以盛放雨水的数组“坑位”,都是前边数比后边若干数小,
//遍历数组,找到每个位置的左右边最大值,其所盛放的水量就是两边最大值的较小值-该位置高度
int len=height.length;
if(len==0)
return 0;
int ans=0;
//改进,用空间换时间
//建立两个数组,分别存放某位置元素的左边最大值和右边最大值
int[] lM=new int[len];
int[] rM=new int[len];
lM[0]=height[0];
rM[len-1]=height[len-1];
for(int i=1;i<len;i++)
lM[i]=Math.max(height[i],lM[i-1]);
for(int i=len-2;i>=0;i--)
rM[i]=Math.max(height[i],rM[i+1]);
for(int i=1;i<len-1;i++) {
int a=Math.min(rM[i], lM[i])-height[i];
if(a>0)
ans+=a;//
}
return ans;