[力扣上岸] 如何用贪心算法解出大厂爱考的接雨水问题?

Posted GTAlgorithm

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[力扣上岸] 如何用贪心算法解出大厂爱考的接雨水问题?相关的知识,希望对你有一定的参考价值。

点击上方“ 力扣上岸 ”,选择“ 设为星标
第一时间关注技术干货!

前言

现在流行的趋势是大厂越来越喜欢考算法题。接雨水问题是一道常考题目,但是却是力扣上一道困难题。能够现场写出时间复杂度为O(n)的算法无疑是一大加分项!一起来看题吧!

题目

思路

假设height数组的长度为n,我们要找出来的答案就是从索引0到索引n-1上面每一个元素能够接到的雨水的最大值的和。因此,此题具有贪心算法的重要特征:用局部最优解求出总体最优解

贪心算法一般是这样的模版:

总结果初始化
while 目标未完成:
  解决第i个子问题
  i加一
  第i个子问题的结果放入总结果中
返回总结果


那么,每一个位置最多能够接多高的雨水呢?我们知道一个理论叫做木桶理论,即木桶能装的水由最短的木板决定。对于一个数组,我们可以理解为一个二维的“木桶”:只有左右两边各一个木板。“木板“对应本题中的左右两边的最大的数字。每一个位置能装的最高的雨水,由左右两边的最大数字中的较小的那个值决定。每一个位置都是如此所以,左右两个“木板”的高度由下面这个公式算出:

left_max = max(left_max, height[left])
right_max = max(right_max, height[right])


这个地方用left和right两个指针是因为我们要计算左右两个木板。

题解

class Solution:
    def trap(self, height: List[int]) -> int:
        if not height: return 0
        n = len(height)
        leftright = 0, n-1
        lmax = height[left]
        rmax = height[right]
        res = 0
        while left <= right:
            if lmax < rmax:
                lmax = max(lmax, height[left])
                res += lmax - height[left]
                left += 1
            else:
                rmax = max(rmax, height[right])
                res += rmax - height[right]
                right -= 1
        return res


需要注意的是:
  • 终止条件是小于等于而不是小于,因为循环中的每一步都只是计算了left right两个指针中的其中一个,如果最后一步left依然小于right,那么会少计算一个元素。

  • 把每一个子问题的结果放入总结果的时候,记得要减掉当前指针对应元素的高度。

时间复杂度只有O(n)!这就是双指针+动态规划的魅力。

精彩回顾





最后

如果文章和笔记能带给你一丝帮助或者启发,请不要吝啬你的赞和在看,你的肯定是我前进的最大动力

以上是关于[力扣上岸] 如何用贪心算法解出大厂爱考的接雨水问题?的主要内容,如果未能解决你的问题,请参考以下文章

❤️思维导图整理大厂面试高频数组12: 4种方法彻底解决接雨水问题, 力扣42❤️

❤️思维导图整理大厂面试高频数组17: 股票问题I的通用dp数组定义法 和 贪心法思想, 力扣121❤️

❤️思维导图整理大厂面试高频数组17: 股票问题I的通用dp数组定义法 和 贪心法思想, 力扣121❤️

力扣算法题—042接雨水

算法| 贪心算法:如何用贪心算法实现Huffman压缩编码?

零基础到秋招上岸互联网大厂算法岗经验