给定一个直方图(也称柱状图),假设有人从上面源源不断地倒水,最后直方图能存多少水量?直方图的宽度为 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
题解:参考大神写的
某一个柱子能否存储水,判断的准则是左侧柱子最大高度、右侧柱子最大高度的较小值是否高于柱子本身,差值为正即可存储水,避免使用辅助空间,先找到数组中的最高点。该点将数组分为左右两半。对于左半侧,从左向右处理;对于右半侧,从右向左处理。处理的时候只需考虑已扫描一侧的高度即可,因另一侧有数组最高点,无需担心蓄水问题。
public class Solution {
public int trap(int[] height) {
if(height==null||height.length<3){
return 0;
}
int n=height.length;
//寻找最高的板子
int highest=0;
int highestIndex=0;
for(int i=0;i<n;i++){
if(highest<height[i]){
highest=height[i];
highestIndex=i;
}
}
int total=0;
//从左边找相对较高的位置
for(int i=0,top=0;i<highestIndex;i++){
if(height[i]>=height[top]){
//左侧相对相告的位置
top=i;
}else{
total+=height[top]-height[i];
}
}
for(int i=n-1,top=n-1;i>highestIndex;i--){
if(height[i]>=height[top]){
top=i;
}else{
total+=height[top]-height[i];
}
}
return total;
}
}