leetcode 407. 接雨水 II BFS

给定一个m x n的矩阵，其中的值均为正整数，代表二维高度图每个单元的高度，请计算图中形状最多能接多少体积的雨水。
说明:
m 和 n 都是小于110的整数。每一个单位的高度都大于0 且小于 20000。
示例：
给出如下 3x6 的高度图:
[
[1,4,3,1,3,2],
[3,2,1,3,2,4],
[2,3,3,2,3,1]
]
返回 4。

如上图所示，这是下雨前的高度图[[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]] 的状态。

下雨后，雨水将会被存储在这些方块中。总的接雨水量是4。

思路：利用最小优先队列，每次最先访问高度最低的方块x，遍历他上下左右四个方向，若某个方块y符合条件，那么
结果就直接加上他们之间的高度差，并把x的高变成为y的高度即可(y的高一定是包围x四周的高的最低长度)。

class Solution {
    int[] dx={0,0,1,-1};
    int[] dy={1,-1,0,0};
    static class Idx implements Comparable<Idx>,Comparator<Idx>{
        int h;
        int x;
        int y;
        public Idx(int x, int y, int h) {
            this.x = x;
            this.y = y;
            this.h = h;
        }
        @Override
        public int compareTo(Idx idx) {
            return h-idx.h;
        }
        @Override
        public int compare(Idx idx, Idx t1) { 
            return idx.h-t1.h; //设置最小优先队列的条件 以h为基准
        }
    }
    public int trapRainWater(int[][] heightMap) {
        if(heightMap.length==0||heightMap[0].length==0){
            return 0;
        }
        int n=heightMap.length;
        int m=heightMap[0].length;
        boolean[][] vis=new boolean[n][m];
        PriorityQueue<Idx> queue=new PriorityQueue<Idx>(); //smallest priority queue
        for(int i=0;i<m;++i){ //四周一定是不能注水的 所以把四周作为初始的访问对象
            vis[0][i]=true;
            queue.add(new Idx(0,i,heightMap[0][i]));
            vis[n-1][i]=true;
            queue.add(new Idx(n-1,i,heightMap[n-1][i]));
        }
        for(int i=1;i<n-1;++i){
            vis[i][0]=true;
            queue.add(new Idx(i,0,heightMap[i][0]));
            vis[i][m-1]=true;
            queue.add(new Idx(i,m-1,heightMap[i][m-1]));
        }
        int ans=0;
        while (!queue.isEmpty()){
            Idx cur=queue.poll();
            for(int i=0;i<4;++i){
                int xi=cur.x+dx[i];
                int yi=cur.y+dy[i];
                if(xi>=0&&yi>=0&&xi<n&&yi<m&&!vis[xi][yi]){
                    ans+=Math.max(0,cur.h-heightMap[xi][yi]); //取高度差
                    queue.add(new Idx(xi,yi,Math.max(cur.h,heightMap[xi][yi])));
                    //把当前高度设置为把他高的方块的高
                    vis[xi][yi]=true;
                }
            }
        }
        return ans;
    }
}