[백준] 1717. 집합의 표현 (Python)

link 🔗 https://www.acmicpc.net/problem/1717

1717번: 집합의 표현

 

💬 유니온 파인드 문제!

💬 유니온 파인드를 오랜만에 풀어서, 개념부터 다시 봐야 했다ㅠㅠ

💬 type이 1일 경우 union 함수 내에서 YES 혹은 NO를 구분한다.

 

import sys
input = sys.stdin.readline

def find(n):
    if parents[n] == n:
        return n
    parents[n] = find(parents[n])
    return parents[n]

def union(type, n1, n2):
    fn1 = find(n1)
    fn2 = find(n2)
    if type == 1:
        if fn1 == fn2:
            print("YES")
        else:
            print("NO")
    else:
        parents[fn2] = fn1

N, M = map(int, input().split())

parents = [num for num in range(N+1)]

for _ in range(M):
    type, num1, num2 = map(int, input().split())
    union(type, num1, num2)