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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
代码

/**
	   * 思路：根据“木桶原理”，决定能盛多少水永远是最短的木板，该题也一样。
	 *    暴力思路：选取height[i]的一个点，分别往左右两端扩散，分别找到改height[i]
	 *    	     的左边最大left_max和右边最大right_max,由于决定水多少的是较短的一方，
	 *    	     所以选取 min(left_max,right_max),但由于与自身height[i]也会占用空间，
	 *    	     所以需要减去height[i];这样一来，该点所能盛的水量为  min(left_max,right_max)-height[i],
	 *    	     循环，不断求取每个点的盛水容量，累加即可；
	 * @param height
	 * @return
	 */
	public int trap(int[] height) {
		int size = height.length;
		int ans = 0;
		for (int i = 1; i < size - 1; i++) {
			int left_max = 0, right_max = 0;
			for (int j = i; j >= 0; j--) {
				left_max = Math.max(left_max, height[j]);
			}
			for (int j = i; j < size; j++) {
				right_max = Math.max(right_max, height[j]);
			}
			ans += Math.min(left_max, right_max) - height[i];
		}
		return ans;
	}

/**
	   * 思路：延续暴力的思路，只不过我们可以发现，当我们每次求一个点的盛水量的时候，都需要求两边的最大值
	   *            这会造成时间复杂度的升高，所以我们可以利用动态规划的思路，用数组存起来。 
	 * @param height
	 * @return
	 */
	public int trap(int[] height) {
		int n=height.length;
		int[] left_max=new int[n];
		int[] right_max=new int[n];
		
		left_max[0]=height[0];
		for(int i=1;i<n;i++) {
			left_max[i]=Math.max(left_max[i-1],height[i]);
		}
		right_max[n-1]=height[n-1];
		for(int i=n-2;i>=0;i--) {
			right_max[i]=Math.max(right_max[i+1],height[i]);
		}
		int ans=0;
		for(int i=0;i<n;i++) {
			ans+=Math.min(left_max[i],right_max[i])-height[i];
		}
		return ans;
 	}

/**
	   * 单调栈：以当前height[i]为起点往后看，如果高度比height[i]低的，根本无法储水；
	   * 	只有比height[i]高的才能储水
	 * @param height
	 * @return
	 */
	public int trap(int[] height) {
		int ans=0;//	记录水面积
		int current=0;//	索引
		Stack<Integer> stack=new Stack<>();
		while(current<height.length) {
			while(!stack.isEmpty()&&height[current]>height[stack.peek()]) {
				Integer pop = stack.pop();
				if(stack.isEmpty()) break;
				int distance=current-stack.peek()-1;//	求水的宽度
				int h=Math.min(height[current],height[stack.peek()])-height[pop];//水的高度
				ans+=distance*h;//面积累加
			}
			stack.push(current++);
		}
		return ans;
 	}
	

标签： 单调栈 数组 双指针