面试题 17.21. 直方图的水量

面试题 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); 不需要额外内存空间。