LeetCode重点题系列之【42. 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.

LeetCode重点题系列之【42. Trapping Rain Water】
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!

Example:

Input: [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6

This problem is very hard. First, you should scan from left to right, keep track current largest height. Second, scan for the right to left, keep tracking the current largest height. Maintain a result list, use ith element in the minimal height of left to right list and right to the left list to minus ith element in height. Return sum of res.

class Solution(object):
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        if len(height)<3:
            return 0
        left2right=[0 for _ in range(len(height))]
        right2left=[0 for _ in range(len(height))]
        left2right[0]=height[0]
        right2left[len(height) - 1] = height[len(height) - 1]
        for i in range(1,len(height)):
            if height[i]>left2right[i-1]:
                left2right[i]=height[i]
            else:
                left2right[i]=left2right[i-1]
        for i in range(len(height)-2,-1,-1):
            if height[i]>right2left[i+1]:
                right2left[i]=height[i]
            else:
                right2left[i]=right2left[i+1]
        res=[]
        for i,item in enumerate(height):
            res.append(min(left2right[i],right2left[i])-item)
        return sum(res)
        

 

上一篇:闪电暴雨


下一篇:前端面试百分之九十九过的技巧(二)