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

【leetcode】Trapping Rain Water

时间:2014-07-18 11:38:45      阅读:228      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   strong   os   

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!


 

题解:在网上研究了半天,才找到时间O(n)和空间O(1)的算法:

  1. 首先找出最大的bar,索引记录在totalMaxIndex这个变量里面;
  2. 从左往右,依次处理totalMaxIndex左边的bar;设置一个curMax记录遍历到当前为止最大的bar,那么对于A[i],有两种可能:一是A[i]<curMax,那么这时候就可以储水(curMax - A[i]);而是A[i] > curMax,这时更新curMax = A[i];
  3. 从右往左,依次处理totalMaxIndex右边的bar,方法同2.

具体算法过程如下图所示:

bubuko.com,布布扣

基本的原理就是当从左往右遍历的时候,最高的bar保证了能够在右端将水拦住,那么就只要计算对于A[i],在左端能够拦截住多高的水就可以了,即变量curMax。

代码如下:

 1 public class Solution {
 2     public int trap(int[] A) {
 3         if(A == null || A.length == 0)
 4             return 0;
 5         
 6         int curMax = 0;
 7         int totalMax = A[0];
 8         int totalMaxIndex = 0;
 9         int answer = 0;
10         
11         for(int i = 1;i < A.length;i++){
12             if(A[i] > totalMax){
13                 totalMax = A[i];
14                 totalMaxIndex = i;
15             }
16         }
17         
18         for(int i = 0;i < totalMaxIndex;i++){
19             if(A[i] < curMax)
20                 answer += curMax - A[i];
21             else
22                 curMax = A[i];
23         }
24         
25         curMax = 0;
26         for(int i = A.length-1;i > totalMaxIndex;i--){
27             if(A[i] < curMax)
28                 answer += curMax - A[i];
29             else {
30                 curMax = A[i];
31             }
32         }
33         
34         return answer;
35     }
36 }

【leetcode】Trapping Rain Water,布布扣,bubuko.com

【leetcode】Trapping Rain Water

标签:style   blog   http   color   strong   os   

原文地址:http://www.cnblogs.com/sunshineatnoon/p/3853029.html

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