面试金典--面试题 17.21. 直方图的水量(不困难的困难题)

文章目录

题目描述

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

上面是由数组 0,1,0,2,1,0,1,3,2,1,2,1 表示的直方图,在这种情况下,可以接 6 个单位的水(蓝色部分表示水)。
示例:

输入: 0,1,0,2,1,0,1,3,2,1,2,1

输出: 6

思路分析

观察图片可知,总面积=黑色面积+蓝色面积。

黑色面积很好求,直接sum(所给数组)就行了。

蓝色面积是答案。所以问题转化成求图中的总面积。

这个其实也很好处理,我们一层一层的去求就好了。

按照所给示例:height=0,1,0,2,1,0,1,3,2,1,2,1
第一层:

left=0,right=11,high=1 (high代表当前层数)。当左右指针指向的区域高度小于high时,左右指针都向中间移动,直到指针指向区域大于等于high的值。若不小于high,则指针不移动。

left不断向右靠近,在第一层left=1时,left和right在输入数组height中的数值都大于当前遍历的层数high。所以第一层的体积就是right-left+1

第二层:

第二层,high = 2,left一直向右移动到left = 3,right向左移动到right = 10。所以第二层体积:right - left + 1 = 8。

第三层同理为1。

所以最后的答案就是,刚求得的总体积-所给数组的和。

完整代码

python 复制代码
class Solution:
    def trap(self, height: List[int]) -> int:
        if not height:
            return 0
        zhuzi = sum(height)
        length = 1 # 表示第几层
        res = 0
        while length<=max(height):
            left = 0
            right = len(height)-1
            while height[left]<length:
                left+=1
            while height[right]<length:
                right-=1
            res += right-left+1
            length+=1
        return res-zhuzi
相关推荐
JieE21231 分钟前
LeetCode 56. 合并区间|超清晰 JS 图解思路,面试高频区间题
javascript·算法·面试
JustHappy10 小时前
我汇总了身边朋友的经历才发现,其实第一份实习是最难找的......
前端·后端·面试
uhakadotcom10 小时前
在python 的 工程化架构中 ,什么是 薄包装器层?
后端·面试·github
假如让我当三天老蒯14 小时前
模块化:ES Module 与 CommonJS 的区别
前端·面试
沉默王二14 小时前
面试官:RAG 不用向量数据库,用 MySQL 硬扛?我:100 万向量不是很轻松?
mysql·面试·ai编程
Darling噜啦啦1 天前
列表转树算法深度解析:从 Map 到 Reduce 的两种实现,面试高频考点
数据结构·算法·面试
swipe1 天前
正则表达式入门到进阶:从表单校验到手写模板引擎
前端·javascript·面试
神奇小汤圆1 天前
RAG大厂面试题汇总:向量检索、混合检索、Rerank、幻觉处理高频问题
面试
假如让我当三天老蒯1 天前
回归基本功:Map/Set 与 WeakMap/WeakSet 的区别
前端·面试