蓝桥杯 python并查集

并查集

并查集是一种树型的数据结构，用于处理一些不相交集合的合并及查询问题

并查集的思想是用一个数组表示了整片森林，树的根节点唯一标识了一个集合，我们只要找到了某个元素的树根，就能确定它在哪个集合里。

n,m=map(int,input().split())
p=[i for i in range(n+1)]
def root(x):
    if p[x]!=x:
        p[x]=root(p[x])
    return p[x]
def union(x,y):
    if root(x)!=root(y):
        p[root(x)]=root(y)
for i in range(m):
    z,x,y=map(int,input().split())
    if z==1:
        union(x,y)
    else:
        if root(x)==root(y):
            print('YES')
        else:
            pri