[백준] 1717번 집합의 표현 파이썬 코드

코드

import sys
sys.setrecursionlimit(10**5) # 재귀 깊이 제한
input = sys.stdin.readline # 빠른 입력

n,m = map(int, input().split())
set = [i for i in range(n+1)] # 부모 노드를 기록할 리스트

def find(a, parent):
  if a != parent[a]: # 루트 노드가 아니라면
    parent[a] = find(parent[a], parent) # 루트 노드에 도달할 때까지 순회
  return parent[a] # 루트 노드 반환

def union(a,b,parent): # 트리 합치기
  a = find(a, parent) # 루트 노드 찾기
  b = find(b, parent)
  if a < b: # 일반적으로 작은 값이 부모 노드로 만든다.
    parent[b] = a
  else:
    parent[a] = b

for _ in range(m):
  o,a,b = map(int, input().split())
  if o == 0:
    union(a,b,set)
  elif o == 1:
    if find(a,set) == find(b,set):
      print("YES")
    else: print("NO")

서로소 집합에 대한 개념을 알 수 있는 문제였다.