首页 > 移动平台 > 详细

[leetcode] Trapping rain water

时间:2016-08-27 07:33:56      阅读:182      评论:0      收藏:0      [点我收藏+]

42. 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.

 

Solution:

Sum water amount of each bin (width=1). Maintain a height of left and right seperately, which is like a one-side wall of partial container. Fix the higher one and add the water amount regard to the lower part. 

 

class Solution(object):
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        left, right = 0, 0
        start, end = 0, len(height)-1
        result = 0
        while start <= end:
            if left<right:
                if height[start]>=left:
                    left = height[start]
                else:
                    result += left - height[start]
                start += 1
            else:
                if height[end]>=right:
                    right = height[end]
                else:
                    result += right - height[end]
                end -= 1
        return result

 

[leetcode] Trapping rain water

原文:http://www.cnblogs.com/chrisyao/p/5812199.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!