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

题目描述

在一个 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 更高的，所以你可以游向坐标方格中的任意位置
示例2:
输入: [[0,1,2,3,4],[24,23,22,21,5],[12,13,14,15,16],[11,17,18,19,20],[10,9,8,7,6]]
输出: 16
解释:
0 1 2 3 4
24 23 22 21 5
12 13 14 15 16
11 17 18 19 20
10 9 8 7 6
最终的路线用加粗进行了标记。
我们必须等到时间为 16，此时才能保证平台 (0, 0) 和 (4, 4) 是连通的
提示:
2 <= N <= 50.
grid[i][j] 是 [0, …, N*N - 1] 的排列。

算法分析

1. 并查集
   题目要我们找的是最少等待时间，可以把网格抽象成一个无权图，每经过一个时刻，依次增加水位高度,考虑此时和水位高度相等的单元格，考虑这个单元格的上、下、左、右、四个方向，并且高度更低的单元格，把它们在并查集中进行合并。

2. 二分法
   使用二分查找定位最短等待时间。具体来说：在区间 [0, N * N - 1] 里猜一个整数，针对这个整数从起点（左上角）开始做一次深度优先遍历或者广度优先遍历。
   当小于等于该数值时，如果存在一条从左上角到右下角的路径，说明答案可能是这个数值，也可能更小；
   当小于等于该数值时，如果不存在一条从左上角到右下角的路径，说明答案一定比这个数值更大。
   按照这种方式不断缩小搜索的区间，最终找到最少等待时间。

代码注释

1. 并查集

public class Solution {

    private int N;//矩阵长宽

    public static final int[][] DIRECTIONS = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};//可以游动的四个方向

    public int swimInWater(int[][] grid) {
        this.N = grid.length;

        int len = N * N;//结点数量
        int[] index = new int[len];//图的邻接矩阵
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                index[grid[i][j]] = getIndex(i, j);
            }
        }

        UnionFind unionFind = new UnionFind(len);
        for (int i = 0; i < len; i++) {
            int x = index[i] / N;
            int y = index[i] % N;//获得当前序号结点在矩阵中的位置

            for (int[] direction : DIRECTIONS) {
                int newX = x + direction[0];
                int newY = y + direction[1];//四个方向依次试探
                if (inArea(newX, newY) && grid[newX][newY] <= i) {
                    unionFind.union(index[i], getIndex(newX, newY));//将新位置与原位置连通
                }

                if (unionFind.isConnected(0, len - 1)) {
                    return i;//返回结果
                }
            }
        }
        return -1;
    }

    private int getIndex(int x, int y) {
        return x * N + y;//根据点在矩阵中的位置获得结点序号
    }

    private boolean inArea(int x, int y) {
        return x >= 0 && x < N && y >= 0 && y < N;//判断当前结点是否在矩阵范围内
    }

    private class UnionFind {

        private int[] parent;//表示并查集中结点关系的数组

        public UnionFind(int n) {
            this.parent = new int[n];
            for (int i = 0; i < n; i++) {
                parent[i] = i;//初始化parent数组
            }
        }

        public int root(int x) {
            while (x != parent[x]) {
                parent[x] = parent[parent[x]];
                x = parent[x];
            }
            return x;//寻找x的父节点
        }

        public boolean isConnected(int x, int y) {
            return root(x) == root(y);//判断x与y是否在同一个集合中
        }

        public void union(int p, int q) {
            if (isConnected(p, q)) {
                return;//若p与q已经在同一个集合中，无需合并直接返回
            }
            parent[root(p)] = root(q);//将p连接在q上
        }
    }
}

2. 二分法

public class Solution {

    private int N;//邻接矩阵长宽

    public static final int[][] DIRECTIONS = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};//可以游动的四个方向

    public int swimInWater(int[][] grid) {
        this.N = grid.length;

        int left = 0;//左边界
        int right = N * N - 1;//右边界
        while (left < right) {
            // left + right 不会溢出
            int mid = (left + right) / 2;
            boolean[][] visited = new boolean[N][N];
            if (grid[0][0] <= mid && dfs(grid, 0, 0, visited, mid)) {
                right = mid;//在当前mid下存在从起点到终点的路径，考虑减小mid值
            } else {
                left = mid + 1;//在当前mid下不存在从起点到终点的路径，考虑增大mid值
            }
        }
        return left;
    }

    
    private boolean dfs(int[][] grid, int x, int y, boolean[][] visited, int threshold) {
        visited[x][y] = true;
        for (int[] direction : DIRECTIONS) {
            int newX = x + direction[0];
            int newY = y + direction[1];//试探四个方向
            if (inArea(newX, newY) && !visited[newX][newY] && grid[newX][newY] <= threshold) {
                if (newX == N - 1 && newY == N - 1) {
                    return true;//遍历到终点
                }

                if (dfs(grid, newX, newY, visited, threshold)) {
                    return true;//递归
                }
            }
        }
        return false;
    }

    private boolean inArea(int x, int y) {
        return x >= 0 && x < N && y >= 0 && y < N;//判断当前位置是否在邻接矩阵中
    }
}

题目总结

和昨天的题没有太大区别,巩固并查集的应用