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

【Trapping Rain Water】cpp

时间:2015-04-26 10:49:01      阅读:135      评论:0      收藏:0      [点我收藏+]

标签:

题目

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.

技术分享

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!

代码

class Solution {
public:
    int trap(vector<int>& height) {
            // non valid input
            int len = height.size();
            if (len<=2) return 0;
            // initial
            int left_max[height.size()];
            int right_max[height.size()];
            left_max[0] = 0;
            right_max[len-1] = 0;
            // get left_max and right_max
            for (int i = 1; i < len; ++i)
            {
                left_max[i] = std::max(left_max[i-1], height[i-1]);
                right_max[len-i-1] = std::max(right_max[len-i], height[len-i]);
            }
            // calculate the sum
            int sum = 0;
            for (int i = 0; i < len; ++i)
            {
                int h = std::min(left_max[i], right_max[i]);
                if (h>height[i])
                {
                    sum += h-height[i];
                }
            }
            return sum;
    }
};

Tips:

1. 遍历,获得每个位置上左边最高的和右边最高的;选择左边和右边比较小的高度,减去该位置的高度,就是可需水量。

2. 注意一些极端case的处理

【Trapping Rain Water】cpp

标签:

原文地址:http://www.cnblogs.com/xbf9xbf/p/4457356.html

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