力扣 leetode 778. 水位上升的泳池中游泳 (python)

Topic

在一个 N x N 的坐标方格 grid 中，每一个方格的值 grid[i][j] 表示在位置 (i,j) 的平台高度。
现在开始下雨了。当时间为 t 时，此时雨水导致水池中任意位置的水位为 t 。你可以从一个平台游向四周相邻的任意一个平台，但是前提是此时水位必须同时淹没这两个平台。假定你可以瞬间移动无限距离，也就是默认在方格内部游动是不耗时的。当然，在你游泳的时候你必须待在坐标方格里面。
你从坐标方格的左上平台 (0，0) 出发。最少耗时多久你才能到达坐标方格的右下平台 (N-1, N-1)？

Example_1

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

Example_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) 是连通的

Tips

2 <= N <= 50.
grid[i][j] 是 [0, …, N*N - 1] 的排列。

Solution

本思路与昨天题目思路相似
可以参考leetcode 1631. 最小体力消耗路径
并查集代码基于 基于列表的并查集实现模板实现
Kruskal算法详解

本题我们还是运用类Kruskal算法
整体思路上我们利用并查集
来判断第一个点和最后一个点是否连通
从而判断能否到达

每当水位上升一
我们就有一些点可以到达

第一步 计算权重

在本题中权重可以类似于1631题
将其看作相邻位置平台水位的较大值
之后将grid变为[边的开始节点, 边的结束节点， 权重]的形式

第二步 按照权重排序

按照权重从小到大排序

第三步 并入边

初始化一个并查集uf
将边按权重从小到大顺序连通
如果第一个节点与最后一个节点连通
则输出当前边的权重即为结果

Code

class UnionFind:
    def __init__(self, n):
        self.father = list(range(n))
        self.size =