1717 집합의 표현

 

union find 연습문제로 풀었던 거

 

#include <iostream>
#include <vector>

using namespace std;

vector<int> parent;

void    init(int n)
{
    parent.resize(n + 1);

    for (int i = 0; i <= n; i++)
        parent[i] = i;
}

int     find(int x)
{
    if (x == parent[x])
        return x;
    return (parent[x] = find(parent[x]));
}

void    merge(int x, int y)
{
    x = find(x);
    y = find(y);
    
    parent[x] = y;
}

int     main()
{
    int n, m;

    scanf("%d %d", &n, &m);
    init(n);
    for (int i = 0; i < m; i++)
    {
        int num, a, b;
        scanf("%d %d %d", &num, &a, &b);
        if (num == 0)
        {
            merge(a, b);
        }
        else
        {
            if (find(a) == find(b))
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    
    return (0);
}