代码随想录算法训练营day54| 卡码网107.寻找存在的路径

学习资料：https://www.programmercarl.com/kamacoder/图论并查集理论基础.html#背景

并查集的两个功能：
1、将两个元素添加到一个集合
2、判断两个元素是否在同一个集合（节点具有同一个根，则在一个集合）
套路：并查集初始化；节点寻根；两节点连线；是否在同一集合（根相同）

学习记录：
卡码网107.寻找存在的路径

点击查看代码

# 判断两个节点是否在同一个集合中：并查集解法，两个节点的根一样，则在一个集合中
# 并查集套路：初始化；寻找根节点；将两个节点连线

class UnionFind:
    def __init__(self, size):
        self.parent = list(range(size+1)) # 初始化并查集
    def find(self, u):
        """找节点u的根节点"""
        if self.parent[u] != u:
            self.parent[u] = self.find(self.parent[u])   # 路径压缩
        return self.parent[u]
        
    def union(self, u, v):
        """给u和v连线"""
        root_u = self.find(u)
        root_v = self.find(v)
        if root_u != root_v:
            self.parent[root_v] = root_u
    
    def is_same(self, u, v):
        """其实是判断两个节点的根是否一样，如果一样，就有连线就有路"""
        return self.find(u) == self.find(v)
        

# 输入值
# 几个节点，几条边
n,m = map(int, input().split())
# 边的信息
uf = UnionFind(n)
for _ in range(m):
    s, t = map(int, input().split())
    uf.union(s, t)
# 起点和终点
source, destination = map(int, input().split())

# 如果起点和终点一样，则路径存在
if uf.is_same(source, destination):
    print(1)
else:
    print(0)