[백준] 1717번 집합의 표현 (파이썬)

1717번: 집합의 표현

 

 

코드

import sys
sys. setrecursionlimit(10**6)

n, m = map(int, sys.stdin.readline().split(" "))
li = [0] * (n+1)
answer_li = []

for i in range(n+1):
    li[i] = i

def find(a):
    if a == li[a]:
        return a
    p = find(li[a])
    li[a] = p
    return p

def union(x,y):
    a = find(x)
    b = find(y)

    if a == b:
        return
    li[max(a,b)] = min(a,b)


for _ in range(m):
    u, a, b = map(int, sys.stdin.readline().split(" "))
    if u == 0:
        union(a,b)
    else:
        if find(a) == find(b):
            answer_li.append("YES")
        else:
            answer_li.append("NO")

for i in range(len(answer_li)):
    print(answer_li[i])