LeetCode 42. 接雨水（4.4打卡）

题目

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

代码（又是抄作业的一天，唉，太菜了）

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;