42. Trapping Rain Water

https://leetcode.com/problems/trapping-rain-water/description/
题意:接雨水
思路1:暴力。虽然暴力,但很巧妙。i遍历一次数组,每次以i为中心,用指针j向两侧遍历,找左右最高的数maxleft和maxright,则i处能留下的雨水最多为:左右max的最小值减去i处的高度。
复杂度o(n^2),c++没超时,py却超时了

class Solution:
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        n = len(height)
        if n < 2:
            return 0
        water = 0
        #对每个格子遍历
        for i in range(n):
            maxLeft, maxRight = 0, 0
            for j in range(i, -1, -1):  #找i左边最高
                maxLeft = max(height[j], maxLeft)
            for j in range(i, n):  #找i右边最高
                maxRight = max(height[j], maxRight)
            water += min(maxLeft, maxRight) - height[i]  #i处能留下的水量
        return water

思路2:动态规划。这个属于比较简单的动归,用两个长度为n的一维数组leftMax和rightMax记录位置i的左右最大信息。易知i的左侧最大值为height[i]与i-1的leftMax中的较大者,同理i的右侧最大值为height[i]与i+1的rightMax中的较大者。
填好信息表后一次遍历,跟思路1一样每次统计i出的水量并累加即可。复杂度O(n)

class Solution:
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        n = len(height)
        if n < 2:
            return 0
        leftMax, rightMax = [0]*n, [0]*n  #信息表
        leftMax[0] = height[0]  #初始值
        rightMax[n-1] = height[n-1]  #初始值
        water = 0

        for i in range(1, n):  #从左向右填表
            leftMax[i] = max(height[i], leftMax[i-1])
        for i in range(n-2, -1, -1):  #从右向左填表
            rightMax[i] = max(height[i], rightMax[i+1])
        for i in range(n):  #一次遍历
            water += min(leftMax[i], rightMax[i]) - height[i]
        return water
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值