码迷,mamicode.com
首页 > 移动开发 > 详细

LeetCode 41 Trapping Rain Water

时间:2014-11-25 18:42:47      阅读:202      评论:0      收藏:0      [点我收藏+]

标签:java   leetcode   

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

For example, 
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

bubuko.com,布布扣

The above elevation map is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped. Thanks Marcos for contributing this image!

思路:这道题和https://oj.leetcode.com/problems/candy/有点像,分别使用leftmax[A.length]和rightmax[A.length]来存储A[i]左边的最大值和A[i]右边的最大值。
当A[i]<Math.min(leftmax[i], rightmax[i]);时,则第i个位置是可以存储水的,否则就不能存储水。
public class Solution {
	public int trap(int[] A) {
		int result = 0,temp;
		int []leftmax=new int [A.length];
		int []rightmax=new int [A.length];
		for(int i=0,max=-1;i<A.length;i++){
			leftmax[i]=max;
			max=max>A[i]? max:A[i];
		}
		for(int i=A.length-1,max=-1;0<=i;i--){
			rightmax[i]=max;
			max=max>A[i]? max:A[i];	
		}
		for(int i=1;i<A.length-1;i++){
			temp=Math.min(leftmax[i], rightmax[i]);
			if(temp>A[i]){
				result+=temp-A[i];
			}
		}
		return result;
	}
}


LeetCode 41 Trapping Rain Water

标签:java   leetcode   

原文地址:http://blog.csdn.net/mlweixiao/article/details/41483807

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!