并查集

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e5 + 10;
int par[maxn];
int rank[maxn];

void init(int n) {
    for(int i = 0; i < n; i ++) {
        par[i] = i;
        rank[i] = 0;
    }
}

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

int unite(int x, int y) {
    x = find(x);
    y = find(y);
    if(x == y) return;

    if(rank[x] < rank[y])
        par[x] = y;
    else {
        par[y] = x;
        if(rank[x] == rank[y])
            rank[x] ++;
    }
}

bool same(int x, int y) {
    return find(x) == find(y);
}

#include <bits/stdc++.h>
using namespace std;

int N;
int c;
int fa[20];

int dfs(int z) {
    if(fa[z] == z) return z;
    else
        return fa[z] = dfs(fa[z]);
}

int main() {
    scanf("%d", &N);
    for(int i = 1; i <= N; i ++)
        fa[i] = i;
    while(N --) {
        int x, y;
        scanf("%d", &c);
        scanf("%d%d", &x, &y);
        if(c == 0)
            fa[y] = dfs(x);
        else if(c == 1) {
            if(dfs(x) == dfs(y))
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}