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

Trapping Rain Water

时间:2014-07-10 15:00:57      阅读:267      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   strong   width   

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!

 

[解题思路]
对于任何一个坐标,检查其左右的最大坐标,然后相减就是容积。所以,
1. 从左往右扫描一遍,对于每一个坐标,求取左边最大值。
2. 从右往左扫描一遍,对于每一个坐标,求最大右值。
3. 再扫描一遍,求取容积并加和。
#2和#3可以合并成一个循环,

 

 1 class Solution {
 2 public:
 3     int trap(int A[], int n) {
 4         if(n<3)return 0;
 5         int res=0;
 6         vector<int> maxL(n,0);
 7         int maxtemp=A[0];
 8         maxL[0]=maxtemp;
 9         for(int i=1;i<n;i++)
10         {
11             maxL[i]=maxtemp;
12             if(A[i]>maxtemp)
13             maxtemp=A[i];
14         }
15         vector<int> maxR(n,0);
16         maxtemp=A[n-1];
17         maxR[n-1]=maxtemp;
18         for(int i=n-2;i>=0;i--)
19         {
20             maxR[i]=maxtemp;
21             int vol=min(maxR[i],maxL[i])-A[i];
22             if(vol>0)res+=vol;
23             if(A[i]>maxtemp)
24             maxtemp=A[i];
25         }
26         return res;
27     }
28 };

只用一个数组就可以了。

 1 class Solution {
 2 public:
 3     int trap(int A[], int n) {
 4         if(n<3)return 0;
 5         int res=0;
 6         vector<int> maxL(n,0);
 7         int maxtemp=A[0];
 8         maxL[0]=maxtemp;
 9         for(int i=1;i<n;i++)
10         {
11             maxL[i]=maxtemp;
12             if(A[i]>maxtemp)
13             maxtemp=A[i];
14         }
15 
16         maxtemp=A[n-1];
17 
18         for(int i=n-2;i>=0;i--)
19         {
20             int vol=min(maxtemp,maxL[i])-A[i];
21             if(vol>0)res+=vol;
22             if(A[i]>maxtemp)
23             maxtemp=A[i];
24         }
25         return res;
26     }
27 };

 

Trapping Rain Water,布布扣,bubuko.com

Trapping Rain Water

标签:style   blog   http   color   strong   width   

原文地址:http://www.cnblogs.com/hicandyman/p/3835032.html

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