面试题 17.21. 直方图的水量
给定一个直方图(也称柱状图),假设有人从上面源源不断地倒水,最后直方图能存多少水量?直方图的宽度为 1。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的直方图,在这种情况下,可以接 6 个单位的水(蓝色部分表示水).感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/volume-of-histogram-lcci
解题思路
把每个位置单独考虑,每个位置能装的水量对于这个位置前后的最大高度的较小值。
所以只需维护一个左指针指向左侧最大值,然后当到达一个位置之后向右遍历找右侧最大值。
如果min(左侧最大值,右侧最大值)大于该位置的高度,则这个位置可以装水,且装的水量
=min(左侧最大值,右侧最大值)-该位置高度。
代码
Java
class Solution {
public int trap(int[] height) {
int leftMax=0;
int sum =0;
for(int i=0;i<height.length;i++){
int rightMax=0;
for(int j=i+1;j<height.length;j++){
rightMax=Math.max(rightMax , height[j]);
}
if(Math.min(leftMax,rightMax)>height[i]){
sum+=Math.min(leftMax,rightMax)-height[i];
}
leftMax=Math.max(leftMax,height[i]);
}
return sum;
}
}
复杂度分析
时间复杂度:O(n^2); 每次到达一个位置,需要遍历右侧所有位置,遍历次数(n-1)+(n-2)+…+1;
空间复杂度:O(1); 不需要额外内存空间。