778. 水位上升的泳池中游泳

package com.heu.wsq.leetcode.graph;

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

/**
 * 778. 水位上升的泳池中游泳
 * @author wsq
 * @date 2021/1/30
 * 在一个 N x N 的坐标方格 grid 中，每一个方格的值 grid[i][j] 表示在位置 (i,j) 的平台高度。
 * 现在开始下雨了。当时间为 t 时，此时雨水导致水池中任意位置的水位为 t 。你可以从一个平台游向四周相邻的任意一个平台，但是前提是此时水位必须同时淹没这两个平台。假定你可以瞬间移动无限距离，也就是默认在方格内部游动是不耗时的。当然，在你游泳的时候你必须待在坐标方格里面。
 * 你从坐标方格的左上平台 (0，0) 出发。最少耗时多久你才能到达坐标方格的右下平台 (N-1, N-1)？
 *
 * 示例 1:
 * 输入: [[0,2],[1,3]]
 * 输出: 3
 * 解释:
 * 时间为0时，你位于坐标方格的位置为 (0, 0)。
 * 此时你不能游向任意方向，因为四个相邻方向平台的高度都大于当前时间为 0 时的水位。
 * 等时间到达 3 时，你才可以游向平台 (1, 1). 因为此时的水位是 3，坐标方格中的平台没有比水位 3 更高的，所以你可以游向坐标方格中的任意位置
 *
 * 链接：https://leetcode-cn.com/problems/swim-in-rising-water
 */
public class SwimInWater {
    int[][] dirs = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
    public int swimInWater(int[][] grid) {
        int n = grid.length;
        int m = grid[0].length;

        Queue<int[]> pq = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] edge1, int[] edge2) {
                return grid[edge1[0]][edge1[1]] - grid[edge2[0]][edge2[1]];
            }
        });
        pq.offer(new int[]{0, 0});
        // 初始化从原点出发到各个节点的距离
        int[] distTo = new int[n * m];
        Arrays.fill(distTo, Integer.MAX_VALUE);
        distTo[0] = grid[0][0];

        // 保存节点的已遍历情况
        boolean[] seen = new boolean[n * m];
        while (!pq.isEmpty()){
            int[] edge = pq.poll();
            int x = edge[0];
            int y = edge[1];
            int d = edge[2];
            if (seen[x * m + y]){
                continue;
            }
            seen[x * m + y] = true;
            if (x == n - 1 && y == m - 1){
                return distTo[x * m + y];
            }
            for (int[] dir : dirs) {
                int nx = dir[0] + x;
                int ny = dir[1] + y;
                if (nx >= 0 && nx < n && ny >= 0 && ny < m && !seen[nx * m + ny]){
                    distTo[nx * m + ny] = Math.max(distTo[x * m + y], grid[nx][ny]);
                    pq.offer(new int[]{nx, ny});
                }
            }
        }
        return -1;
    }
}