leetcode每日一题—面试题17.21直方图的水量

最新推荐文章于 2023-03-21 11:57:55 发布

柒年时光

最新推荐文章于 2023-03-21 11:57:55 发布

阅读量102

点赞数

分类专栏： leetcode Python

本文链接：https://blog.csdn.net/jqq125/article/details/115448748

版权

Python 同时被 2 个专栏收录

484 篇文章 1 订阅

订阅专栏

leetcode

461 篇文章 1 订阅

订阅专栏

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 1。
在这里插入图片描述

解答：

class Solution:
    def trap(self, height: List[int]) -> int:
        ans = 0
        left, right = 0, len(height) - 1
        leftMax = rightMax = 0

        while left < right:
            leftMax = max(leftMax, height[left])
            rightMax = max(rightMax, height[right])
            if height[left] < height[right]:
                ans += leftMax - height[left]
                left += 1
            else:
                ans += rightMax - height[right]
                right -= 1
        return ans