LeetCode 407. 接雨水 II

题目链接

思路：先将最外层的一圈作为围墙，然后选择高度最小的方块，然后查看该方块的四个相邻方块的高度，如果高度比它大，此方块可删除，用相邻方块组成新的围墙；如果高度比它小，那么相邻方块可储蓄的水为两者高度差，用储水之后的高度增加的相邻方块组成新的围墙。重复上述过程，直到所有的方块都被访问为止。

代码：

class Solution {
    public int trapRainWater(int[][] heightMap) {
        int m = heightMap.length;
        int n = heightMap[0].length;
        int res=0;
        int[][] directions = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}};
        Queue<int[]> q = new PriorityQueue<>((x,y)->x[2]-y[2]);
        boolean[][] visited = new boolean[m][n];
        for(int i=0;i<m;i++){
            q.add(new int[]{i,0,heightMap[i][0]});
            q.add(new int[]{i,n-1,heightMap[i][n-1]});
            visited[i][0]=true;
            visited[i][n-1]=true;
        }
        for(int i=1;i<n-1;i++){
            q.add(new int[]{0,i,heightMap[0][i]});
            q.add(new int[]{m-1,i,heightMap[m-1][i]});
            visited[0][i]=true;
            visited[m-1][i]=true;
        }
        while(!q.isEmpty()){
            int[] p = q.poll();
            for(int[] d:directions){
                int x = p[0]+d[0] ;
                int y = p[1]+d[1];
                if(x<0||x>m-1||y<0||y>n-1||visited[x][y])
                continue;
                if(p[2]>heightMap[x][y]){
                    res+=p[2]-heightMap[x][y];
                    heightMap[x][y]=p[2];
                }
                q.add(new int[]{x,y,heightMap[x][y]});
                visited[x][y]=true;
            }
        }
        return res;
    }
}