Submission #420851

---

Source Code Expand

#include <iostream>
#include <string>
using std::string;

struct State {
    int n;
    int *par;  
};

int root(State &st, int x)
{
    if (st.par[x] == x) {
        return x;
    }
    else {
        return st.par[x] = root(st, st.par[x]);
    }
}

bool same(State &st, int x, int y)
{
    return root(st, x) == root(st, y);
}

void unite(State &st, int x, int y)
{
    x = root(st, x);
    y = root(st, y);
    if (x == y) return;
    st.par[x] = y;
}

int main()
{
    State st;
    int q;
    std::cin >> st.n >> q;
    
    st.par = new int[st.n];
    
    for (int i = 0; i < st.n; i++) {
        st.par[i] = i;
    }
    
    for (int i = 0; i < q; ++i) {
        int p, a, b;
        std::cin >> p >> a >> b;
        if (p == 0) {
            unite(st, a, b);
        }
        else {
            if (same(st, a, b)) {
                std::cout << "Yes" << std::endl;
            }
            else {
                std::cout << "No" << std::endl;
            }
        }
    }
    
    return 0;
}

Submission Info

Judge Result