接雨水2.0

问题描述：
给你一个 m x n 的矩阵，其中的值均为非负整数，代表二维高度图每个单元的高度，请计算图中形状最多能接多少体积的雨水。
示例 1:

输入: heightMap = [[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]]
输出: 4
解释: 下雨后，雨水将会被上图蓝色的方块中。总的接雨水量为1+2+1=4。
示例 2:

输入: heightMap = [[3,3,3,3,3],[3,2,2,2,3],[3,2,1,2,3],[3,2,2,2,3],[3,3,3,3,3]]
输出: 10

问题分析：
堆解题法,从边界最小高度向四周扩散,若遇到比当前高度更小的方块,则记录其高度差,并入堆,同时更新当前最小边界值。

问题求解：

class Solution {
    public int trapRainWater(int[][] heightMap) {

        int m=heightMap.length;
        int n=heightMap[0].length;
        //备忘录
        boolean [][]visited=new boolean[m][n];
        //根据高度构建最小堆
        PriorityQueue<int[]> pq=new PriorityQueue<>((a1,a2)->{
            return a1[2]-a2[2];
        });
        //先把最外围加入最小堆
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (i==0||i==m-1||j==0||j==n-1){
                    pq.offer(new int[]{i,j,heightMap[i][j]});
                    visited[i][j]=true;
                }
            }
        }
        int result=0;
        //出队，往四周方向寻找更小的进行灌水
        while (!pq.isEmpty()){
            int[] poll = pq.poll();
            int x = poll[0];
            int y = poll[1];
            int height = poll[2];
            //左
            if (x-1>=0&&!visited[x-1][y]){
                if (heightMap[x-1][y]<height){
                    //可以灌水
                    result+=height-heightMap[x-1][y];
                    //灌水后的格子成为外围之一
                    pq.offer(new int[]{x-1,y,height});
                }
                else {
                    //没灌水也成为外围
                    pq.offer(new int[]{x - 1, y, heightMap[x - 1][y]});
                }
                visited[x-1][y]=true;
            }
            //右
            if (x+1<m&&!visited[x+1][y]){
                if (heightMap[x+1][y]<height){
                    //可以灌水
                    result+=height-heightMap[x+1][y];
                    //灌水后的格子成为外围之一
                    pq.offer(new int[]{x+1,y,height});
                }else {
                    //没灌水也成为外围
                    pq.offer(new int[]{x + 1, y, heightMap[x + 1][y]});
                }
                visited[x+1][y]=true;
            }
            //下
            if (y-1>=0&&!visited[x][y-1]){
                if (heightMap[x][y-1]<height){
                    //可以灌水
                    result+=height-heightMap[x][y-1];
                    //灌水后的格子成为外围之一
                    pq.offer(new int[]{x,y-1,height});
                }else {
                    //没灌水也成为外围
                    pq.offer(new int[]{x, y - 1, heightMap[x][y - 1]});
                }
                visited[x][y-1]=true;
            }
            //上
            if (y+1<n&&!visited[x][y+1]){
                if (heightMap[x][y+1]<height){
                    //可以灌水
                    result+=height-heightMap[x][y+1];
                    //灌水后的格子成为外围之一
                    pq.offer(new int[]{x,y+1,height});
                }else {
                    //没灌水也成为外围
                    pq.offer(new int[]{x, y + 1, heightMap[x][y + 1]});
                }
                visited[x][y+1]=true;
            }
        }
        return result;

    }
}

问题总结：

老规矩先看官方给出的解题思路。
方法一：最小堆
思路与算法

本题为经典题目，解题的原理和方法都可以参考「42.接雨水」，本题主要从一维数组变成了二维数组。
首先我们思考一下什么样的方块一定可以接住水：

该方块不为最外层的方块；
该方块自身的高度比其上下左右四个相邻的方块接水后的高度都要低；
我们假设方块的索引为 (i,j)(i,j)，方块的高度为 \textit{heightMap}[i][j]heightMap[i][j]，方块接水后的高度为 \textit{water}[i][j]water[i][j]。则我们知道方块 (i,j)(i,j) 的接水后的高度为：

\textit{water}[i][j] = \max(\textit{heightMap}[i],\min(\textit{water}[i-1][j],\textit{water}[i+1][j],\textit{water}[i][j-1],\textit{water}[i][j+1]))
water[i][j]=max(heightMap[i],min(water[i−1][j],water[i+1][j],water[i][j−1],water[i][j+1]))

我们知道方块 (i,j)(i,j) 实际接水的容量计算公式为 \textit{water}[i][j] - \textit{heightMap}[i][j]water[i][j]−heightMap[i][j]。
首先我们可以确定的是，矩阵的最外层的方块接水后的高度就是方块的自身高度，因为最外层的方块无法接水，因此最外层的方块 \textit{water}[i][j] = \textit{heightMap}[i][j]water[i][j]=heightMap[i][j]。

根据木桶原理，接到的雨水的高度由这个容器周围最短的木板来确定的。我们可以知道容器内水的高度取决于最外层高度最低的方块，如图 11 所示：

我们假设已经知道最外层的方块接水后的高度的最小值，则此时我们根据木桶原理，肯定可以确定最小高度方块的相邻方块的接水高度。我们同时更新最外层的方块标记，我们在新的最外层的方块再次找到接水后的高度的最小值，同时确定与其相邻的方块的接水高度，如图 22 所示:

然后再次更新最外层，依次迭代直到求出所有的方块的接水高度，即可知道矩阵中的接水容量。

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/trapping-rain-water-ii/solution/jie-yu-shui-ii-by-leetcode-solution-vlj3/
来源：力扣（LeetCode）