LeetCode 407. 接雨水 II JAVA题解

LeetCode 407. 接雨水 II JAVA题解

题目链接：https://leetcode-cn.com/problems/trapping-rain-water-ii/

根据油管上的一个视频写的java代码，视频链接：
https://www.youtube.com/watch?v=cJayBq38VYw

1. 先将最外围四周看作第一层围栏，矩阵的元素看作节点，将其添加到优先队列中；
2. 依次出队，并进行bfs，过程中维护一个“当前边界最小值”，该值为所有出队的高度中的最大值；
3. 在bfs过程中，即访问出队节点的邻居时，若邻居高度小于“当前边界最小值”，则该邻居节点可储水(“当前边界最小值” - 邻居节点高度)
4. 需要一个visited矩阵，记录bfs遍历过的节点

以下是我写的java代码：

import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Queue;

/**
 * @description:
 * @Author: JachinDo
 * @Date: 2019/11/04 10:44
 */

class TrapNode{
   
     int height;
     int row;
     int col;

    public TrapNode(int height, int row, int col) {
   
        this.height = height;
        this.row = row;
        this.col = col;
    }
}

public class TrapRain2 {
   
    public int trapRainWater(int[][] heightMap) {
   
        int rows = heightMap.length;
        if (rows == 0) {
   
            return 0;
        }
        int cols = heightMap[0].length;
        int[][] visited = new int[rows][cols];
        Queue<TrapNode> priorityQueue = new PriorityQueue<>(new Comparator