[Leetcode][python]Trapping Rain Water/接雨水

【Leetcode题解】 同时被 2 个专栏收录
146 篇文章 0 订阅

题目大意

给定数组A,A[i]表示第i个位置的高度,求可以盛放雨水的容量。
这里写图片描述

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

解题思路

纯思路

网上有多种思路,不尽相同但是思路类似,这里贴一个供参考。

参考:http://www.cnblogs.com/zuoyuan/p/3781453.html

开辟一个数组leftmosthigh,leftmosthigh[i]为A[i]之前的最高的bar值,然后从后面开始遍历,用rightmax来记录从后向前遍历遇到的最大bar值,那么min(leftmosthigh[i], rightmax)-A[i]就是在第i个bar可以储存的水量。例如当i=9时,此时leftmosthigh[9]=3,而rightmax=2,则储水量为2-1=1,依次类推即可。这种方法还是很巧妙的。时间复杂度为O(N)。

https://blog.csdn.net/sunnyyoona/article/details/18557669
代码略

代码

纯思路

class Solution(object):
    def trap(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        left_temp_max = [0 for i in range(len(height))]  # 每个位置的左边最高值
        left_max = 0
        right_max = 0
        for i in range(len(height)):
            if height[i] > left_max: 
                left_max = height[i]
            left_temp_max[i] = left_max

        total = 0
        for i in range(len(height)-1, -1, -1):
            if height[i] > right_max: 
                right_max = height[i]
            if min(right_max, left_temp_max[i]) > height[i]:  # 主要判断
                total += min(right_max, left_temp_max[i]) - height[i]
        return total

总结

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值