[백준 1717번] 집합의 표현

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

1717번: 집합의 표현

2023.04.11 - [코딩 테스트/개념] - Union-Find

# 1 ≤ N ≤ 90

import sys

readline = sys.stdin.readline

n, m = map(int, readline().split())

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


def find(v):
    if union_find[v] == v:
        return v

    new_v = find(union_find[v])
    union_find[v] = new_v
    return new_v


def union(v1, v2):
    find_v1 = find(v1)
    find_v2 = find(v2)

    if find_v1 != find_v2:
        union_find[find_v2] = find_v1


for _ in range(m):
    calc, a, b = map(int, readline().split())

    if calc == 0:
        union(a, b)
    elif calc == 1:
        if find(a) == find(b):
            print("YES")
        else:
            print("NO")

두 집합을 합하고, 같은 집합인지 확인하고.. 유니온 파인드 문제다! 함수 구현했던 거 활용해서 바로 풀었다.