[백준] 집합의 표현

출처::  www.acmicpc.net/problem/1717

분류:: 유니온파인드

 

1. 문제 이해 및 해결과정

 

2. 풀이방법

#집합의 표현
#https://www.acmicpc.net/problem/1717
import sys
sys.stdin = open("input.txt","r")
n,m=map(int,input().split()) #n+1개의 팀, 연산횟수
parent=[]

def find_parent(parent,x):
    if parent[x]!=x:
        parent[x]=find_parent(parent,parent[x])
    return parent[x]

def union_parent(parent,a,b):
    a=find_parent(parent,a)
    b=find_parent(parent,b)
    if a<b:
        parent[b]=a
    else:
        parent[a]=b


for i in range(n+1):
    parent.append(i)

for _ in range(m):
    comm,a,b=map(int,input().split())
    if comm==0: #팀합치기
        union_parent(parent,a,b)
    elif comm==1: #같은팀 여부 확인
        if find_parent(parent,a)==find_parent(parent,b): #부모가 같을 떄, 같은팀
            print("YES")
        else:
            print("NO")

 

3. 오답원인

 

4. 알게된 점