面试前必刷 字节hard 直方图的水量&&接雨水

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 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

---

原来的官解不太好理解，找了一个好理解的

核心思路：总体积减去柱子体积就是水的容量

计算每一层总体积，加和即为总体积

计算过程（图示）

移动指针直到指针区域大于当前区域

第一层

high=1,left向右移动到left = 1，right保持不动。所以第一层体积为volume1 = right - left + 1 = 11，

注意这里需要相减后+1

第二层

high = 2，left一直向右移动到left = 3，right向左移动到right = 10。所以volume2 = right - left + 1 = 8

第三层

不再赘述

public class Solution {
     public int trap(int[] height)
     {
         int sum = 0;
         for(int i = 0;i<height.length;i++) {
            sum += height[i];
         }//求数组总和
         int volume = 0; // 总体积和高度初始化
         int high = 1;
         int size = height.length;
         int left = 0; // 双指针初始化
         int right = size - 1;
         while (left <= right) {
             while (left <= right && height[left] < high) {
                 left++;
             }
             while (left <= right && height[right] < high) {
                 right--;
             }
             volume += right - left + 1; // 每一层的容量都加起来
             high++; // 高度加一
         }
         return volume -sum; // 总体积减去柱子体积，即雨水总量
     }
}

---

参考

作者：threesr
链接：https://leetcode-cn.com/problems/volume-of-histogram-lcci/solution/shuang-zhi-zhen-an-xing-qiu-jie-xiang-xi-d162/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

---

好记版官解力扣

双指针

维护一个leftMax和rightMax，左右指针从两边开始

每次循环刷新最大高度

当前高度哪个低动那个，并加到ans中

循环可加等号可以不加等号

不加等号最后left和right一定汇集到最高点，无雨水

class Solution {
    public int trap(int[] height) {
        int ans = 0;
        int left = 0, right = height.length - 1;
        int leftMax = 0, rightMax = 0;
        while (left < right) {
            leftMax = Math.max(leftMax, height[left]);
            rightMax = Math.max(rightMax, height[right]);
            if (height[left] < height[right]) {
                ans += leftMax - height[left];
                ++left;
            } else {
                ans += rightMax - height[right];
                --right;
            }
        }
        return ans;
    }
}