leetcode每日一题—1631.最小体力的消耗

题目：
你准备参加一场远足活动。给你一个二维 rows x columns 的地图 heights ，其中 heights[row][col] 表示格子 (row, col) 的高度。一开始你在最左上角的格子 (0, 0) ，且你希望去最右下角的格子 (rows-1, columns-1) （注意下标从 0 开始编号）。你每次可以往 上，下，左，右 四个方向之一移动，你想要找到耗费 体力 最小的一条路径。

一条路径耗费的 体力值 是路径上相邻格子之间 高度差绝对值 的 最大值 决定的。请你返回从左上角走到右下角的最小 体力消耗值 。

思路：
1.为每个结点编号，建图，图中每条边的权值为：对应两节点值的绝对值。用三元组列表来记录图中所有的边，并按权值从小到大排序
2.使用并查集：将边依次加入图中，若加入当前边后，起始节点0和终止结点（mn-1)已经联通，则返回当前边的权值。

解答：

# 并查集模板
class UnionFind:
    def __init__(self, n: int):
        self.parent = list(range(n))
        #当前集合的大小
        self.size = [1] * n
        self.n = n
        # 当前连通分量数目
        self.setCount = n
    
    def findset(self, x: int) -> int:
        if self.parent[x] == x:
            return x
        self.parent[x] = self.findset(self.parent[x])
        return self.parent[x]
    
    def unite(self, x: int, y: int) -> bool:
        x, y = self.findset(x), self.findset(y)
        if x == y:
            return False
        if self.size[x] < self.size[y]:
            x, y = y, x
        self.parent[y] = x
        self.size[x] += self.size[y]
        self.setCount -= 1
        return True
    
    def connected(self, x: int, y: int) -> bool:
        x, y = self.findset(x), self.findset(y)
        return x == y

class Solution:
    def minimumEffortPath(self, heights: List[List[int]]) -> int:
        m, n = len(heights), len(heights[0])
        edges = list()
        for i in range(m):
            for j in range(n):
                idx = i * n + j
                if i > 0:
                    edges.append((idx - n, idx, abs(heights[i][j] - heights[i - 1][j])))
                if j > 0:
                    edges.append((idx - 1, idx, abs(heights[i][j] - heights[i][j - 1])))
             
        edges.sort(key=lambda e: e[2])

        uf = UnionFind(m * n)
        ans = 0
        for x, y, v in edges:
            uf.unite(x, y)
            if uf.connected(0, m * n - 1):
                ans = v
                break
        
        return ans