[10-07 그래프] 팀 결성

최근 보고 있는 책인  '이것이 코딩테스트다 with 파이썬 편_나동빈_한빛미디어' 에 있는 문제이다.

www.hanbit.co.kr/store/books/look.php?p_code=B8945183661

이것이 취업을 위한 코딩 테스트다 with 파이썬

 

 

문제는 책 안에 있다!!

 

import sys
input = sys.stdin.readline

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


n,m = map(int,input().split())
parent = [i for i in range(n+1)]

for i in range(m):
    operation, a, b = map(int,input().split())
    if operation == 0:
        union_parent(parent,a,b)
    else :
        if find_parent(parent,a) == find_parent(parent,b) :
            print('YES')
        else :
            print('NO')