仅供自己学习记录
LeetCode刷题日记2021-4-2
题目描述:
给定一个直方图(也称柱状图),假设有人从上面源源不断地倒水,最后直方图能存多少水量?直方图的宽度为 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
题解思路:
- 因为直方图每一层从左边到右边的地方一定会被黑色柱体跟蓝色雨水填满
- 利用左右指针计算出每一层的体积 然后得到总体积
- 用体积减去柱子的体积得到的就是雨水的体积
代码:
def trap(self, height: List[int]) -> int:
#计算柱子体积和
Sum=sum(height)
n=len(height)
#定义左右指针
left,right=0,n-1
#定义总体积跟层高
vol,high=0,1
#计算每层体积 注意体积是右指针-左指针+1 即且左右指针相等时体积为1
while(left<=right):
#左右指针不断移动直到找到与当前层高相等的第一个位置
while(left<=right and height[left]<high):
left+=1
while(left<right and height[right]<high):
right-=1
#左右指针移动到位后 计算当前层数体积 并且层数升高计算下一次体积
vol+=right-left+1
high+=1
return vol-Sum
来源:
https://leetcode-cn.com/problems/volume-of-histogram-lcci/solution/shuang-zhi-zhen-an-xing-qiu-jie-xiang-xi-d162/