Submission #420657

---

Source Code Expand

#include <iostream>
#include <vector>

using namespace std;

int find(vector<int>& v, int c){
    if(v[c] != c){
        v[c] = find(v, v[c]);
    }
    return v[c];
}

void unite(vector<int>& v, int a, int b){
    int ca = find(v, a);
    int cb = find(v, b);
    if(ca != cb) v[ca] = cb;
}

vector<int> init(int n){
    vector<int> v(n);
    for(int i = 0; i < n; i++){
        v[i] = i;
    }
    return v;
}

int main(){
    int n, q;
    cin >> n >> q;
    vector<int> v = init(n);
    for(int i = 0; i < q; i++){
        int p, a, b;
        cin >> p >> a >> b;
        cerr << p << " : " << a << " : " << b << endl;
        if(p == 0){
            unite(v, a, b);
        }else{
            if(find(v, a) == find(v, b)){
                cout << "Yes" << endl;
            }else{
                cout << "No" << endl;
            }
        }
    }
    return 0;
}

Submission Info

Judge Result