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

来源：力扣（LeetCode） 778 水位上升的泳池中游泳

在一个 n x n 的整数矩阵 grid 中，每一个方格的值 grid[i][j] 表示位置 (i, j) 的平台高度。

当开始下雨时，在时间为 t 时，水池中的水位为 t 。你可以从一个平台游向四周相邻的任意一个平台，但是前提是此时水位必须同时淹没这两个平台。假定你可以瞬间移动无限距离，也就是默认在方格内部游动是不耗时的。当然，在你游泳的时候你必须待在坐标方格里面。

你从坐标方格的左上平台 (0，0) 出发。返回 你到达坐标方格的右下平台 (n-1, n-1) 所需的最少时间 。

示例 1:

输入: grid = [[0,2],[1,3]]
输出: 3
解释:
时间为0时，你位于坐标方格的位置为 (0, 0)。
此时你不能游向任意方向，因为四个相邻方向平台的高度都大于当前时间为 0 时的水位。
等时间到达 3 时，你才可以游向平台 (1, 1). 因为此时的水位是 3，坐标方格中的平台没有比水位 3 更高的，所以你可以游向坐标方格中的任意位置
示例 2:

输入: grid = [[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
解释: 最终的路线用加粗进行了标记。
我们必须等到时间为 16，此时才能保证平台 (0, 0) 和 (4, 4) 是连通的
 

提示:

n == grid.length
n == grid[i].length
1 <= n <= 50
0 <= grid[i][j] < n²
grid[i][j] 中每个值 均无重复

如下代码考虑最后一位为最大值的情况，且 max > n * n - 1

方法一：dfs

public static int swimInWater(int[][] grid) {
        int n = grid.length;
        int lo = Math.max(grid[0][0], grid[n - 1][n - 1]);

        int hi = findMax(grid);

        if(grid[n - 1][n - 1] == hi){
            return hi;
        }

        BitSet bs = new BitSet(hi);

        while (lo < hi) {
            int mi = (lo + hi) >> 1;
            if (dfs(grid, 0, 0, mi, bs)) {
                hi = mi;
            } else {
                lo = mi + 1;
            }
            bs.clear();
        }

        return lo;
    }

    /**
     * 查找当钱列表最大值
     * @param grid
     * @return
     */
    public static int findMax(int[][] grid){
        int max = -1;
        for (int[] ints : grid) {
            for (int anInt : ints) {
                max = Math.max(max, anInt);
            }
        }
        return max;
    }

    public static boolean dfs(int[][] grid, int i, int j, int limit, BitSet bs) {
        if (bs.get(i * grid.length + j) || grid[i][j] > limit) {
            return false;
        }
        if (i == grid.length - 1 && j == grid.length - 1) {
            return true;
        }
        bs.set(i * grid.length + j);
        if (i < grid.length - 1 && dfs(grid, i + 1, j, limit, bs)) {
            return true;
        }
        if (j < grid.length - 1 && dfs(grid, i, j + 1, limit, bs)) {
            return true;
        }
        if (i > 0 && dfs(grid, i - 1, j, limit, bs)) {
            return true;
        }
        if (j > 0 && dfs(grid, i, j - 1, limit, bs)) {
            return true;
        }
        return false;
    }


    public static void main(String[] args) {
//        int[][] grid = {{0,5}, {6,3}};
//        int[][] grid = {{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}};
        int[][] grid = {{0, 1, 2, 3, 4}, {24, 23, 22, 21, 5}, {12, 13, 14, 15, 16}, {11, 17, 18, 19, 49}, {10, 9, 8, 47, 50}};
        int res = swimInWater(grid);
        System.out.println(res);
    }

 方法二：并查集

public static void main(String[] args) {
//        int[][] grid = {{0,5}, {6,3}};
        int[][] grid = {{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}};
//        int[][] grid = {{0, 1, 2, 3, 4}, {24, 23, 22, 21, 5}, {12, 13, 14, 15, 16}, {11, 17, 18, 19, 50}, {10, 9, 8, 50, 6}};

        int res = swimInWater(grid);
        System.out.println(res);
    }

    public static int swimInWater(int[][] grid) {
        int n = grid.length, last = n * n - 1;
        List<int[]> list = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                list.add(new int[]{i * n + j, grid[i][j]});
            }
        }

        list.sort(Comparator.comparingInt(l -> l[1]));

        // 最大值在最后一位时直接退出
        if(list.get(last)[1] == grid[n - 1][n - 1]){
            return grid[n - 1][n - 1];
        }

        int[] parents = new int[n * n];
        for (int i = 0; i <= last; i++) {
            parents[i] = i;
        }

        for (int[] ints : list) {
            int x = ints[0] / n, y = ints[0] % n, h = ints[1];

            if(x > 0 && grid[x - 1][y] <= h){
                parents[find(parents, ints[0] - n)] = find(parents, ints[0]);
            }

            if(x < n - 1 && grid[x + 1][y] <= h){
                parents[find(parents, ints[0] + n)] = find(parents, ints[0]);
            }

            if(y > 0 && grid[x][y - 1] <= h){
                parents[find(parents, ints[0] - 1)] = find(parents, ints[0]);
            }
            if(y < n - 1 && grid[x][y + 1] <= h){
                parents[find(parents, ints[0] + 1)] = find(parents, ints[0]);
            }

            if(find(parents, 0) == find(parents, last)){
                return h;
            }
        }


        return -1;
    }

    public static int find(int[] parent, int i) {
        if(parent[i] != i){
            parent[i] = find(parent, parent[i]);
        }
        return parent[i];
    }