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

【LeetCode】Trapping Rain Water

时间:2014-12-15 15:03:44      阅读:184      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   io   ar   color   os   sp   for   

Trapping Rain Water

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!

 

双指针i,j分别从首尾开始扫,记当前i指针遇到的最大值为leftWall,j指针遇到的最大值为rightWall

(1)leftWall <= rightWall

对于i指针指向的位置,被trap的值为(leftWall-A[i])。

i前进一个位置。

解释如下:

a.如果i与j之间不存在比leftWall大的值,那么i位置trap的值就取决与leftWall与rightWall的较小值,也就是leftWall

b.如果i与j之间存在比leftWall大的值,其中离leftWall最近的记为newLeftWall,那么i位置trap的值就取决与leftWall与newLeftWall的较小值,也就是leftWall

(2)leftWall > rightWall

对于j指针指向的位置,被trap的值为(rightWall-A[j])。

j前进一个位置。

解释同上。

 

class Solution {
public:
    int trap(int A[], int n) {
        if(n < 3)   //at least 3 integers
            return 0;
        
        int i = 0;
        int j = n-1;
        int leftWall = 0;
        int rightWall = 0;
        int ret = 0;
        while(i <= j)
        {
            //update leftWall and rightWall
            leftWall = max(leftWall, A[i]);
            rightWall = max(rightWall, A[j]);
            if(leftWall <= rightWall)
            {//no matter whether there is a more higher wall between leftWall and rightWall
             //A[i] is dependent on leftWall if leftWall <= rightWall
                ret += (leftWall-A[i]);
                i ++;
            }
            else
            {//no matter whether there is a more higher wall between leftWall and rightWall
             //A[j] is dependent on rightWall if rightWall < leftWall
                ret += (rightWall-A[j]);
                j --;
            }
        }
        return ret;
    }
};

bubuko.com,布布扣

【LeetCode】Trapping Rain Water

标签:style   blog   http   io   ar   color   os   sp   for   

原文地址:http://www.cnblogs.com/ganganloveu/p/4164763.html

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