Leetcode 42. 接雨水

本文介绍了如何通过贪心策略结合数组的前缀和与后缀和来解决一个关于高度数组的陷阱问题。作者分享了解题思路和Python代码实现,强调了使用premax和postmax数组而非append方法的重要性。
摘要由CSDN通过智能技术生成

在这里插入图片描述

心路历程:

这道题很久之前在看数组前缀的专题的时候见过,是一道典型的用数组前缀最大值来解决问题的题。感觉应该算是贪心思维吧。凭着之前记着的求解思路把这道题写了出来,说实话要是没遇到过这种类型的第一次见的话自己想出来这个贪心解法还是有点困难的。

注意的点:

1、在求后缀最大值时,最好采用post_max[i] = x的赋值方式而不是post_max.append(x)

解法:贪心+数组前后缀最大值

class Solution:
    def trap(self, height: List[int]) -> int:
        # 前缀和和后缀和解决
        n = len(height)
        premax, postmax = [0]*n, [0]*n  # 这块最好别用append方法
        premax_res, postmax_res = height[0], height[-1]
        for i in range(n):
            premax_res = max(premax_res, height[i])
            premax[i] = premax_res
        for i in range(n-1, -1, -1):
            postmax_res = max(postmax_res, height[i])
            postmax[i] = postmax_res
        res = 0
        for i in range(n):
            h = min(premax[i], postmax[i])
            delta_h = max(0, h - height[i])
            res += delta_h * 1
            # print(i, delta_h, premax[i], postmax[i])
        return res
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值