1717 집합의 표현

BOJ::1717 집합의 표현

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

디스조인트셋(disjoint-set)을 활용하는 기본문제이다.

#include <stdio.h>
 
int n, m;
int group[1000001];
 
int find(int a) {
    if (a == group[a]) return a;
    return group[a] = find(group[a]); //
}
 
void merge(int a, int b) {
    a = find(a);
    b = find(b);
    if (a == b) return;
    group[b] = a;
}
 
int main() {
    scanf("%d %d", &n, &m);
    for (int i = 0; i <= n; i++) {
        group[i] = i;
    }
 
    for (int i = 0; i < m; i++) {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        if (a == 0) {
            merge(b, c);
        }
        else {
            if (find(b) == find(c)) printf("YES\n");
            else printf("NO\n");
        }
    }
}