Submission #6903576

---

Source Code Expand

#include <iostream>

#include <vector>
#include <map>
#include <algorithm>
#include <fstream>
#include<cstdio>
#include<iomanip>
#include<stack>
#include<queue>
#include<string>
#include <cstdlib>
#include <typeinfo>
#include <math.h>

#define REP(i, n) for(int i=0;i<n;i++)
#define REP2(i, s, n) for(int i=s;i<n;i++)
#define REP_1(i, n) for(int i=1;i<n+1;i++)
#define bitSearch(bit, n) for(int bit = 0; bit < (1 << n); bit++)
using namespace std;

template<class T>
void print(const T &value) {
    std::cout << value << std::endl;
}

void yesno(bool a) { if (a)cout << "yes" << endl; else cout << "no" << endl; }

void YesNo(bool a) { if (a)cout << "Yes" << endl; else cout << "No" << endl; }

void YESNO(bool a) { if (a)cout << "YES" << endl; else cout << "NO" << endl; }

typedef long long ll;
typedef unsigned long ul;
typedef long double ld;

template<class T>
inline bool chmin(T &a, T b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}

template<class T>
inline bool chmax(T &a, T b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}

ll INF = 10000000;
ll mod = 1000000007;//10^9+7
int dx[8] = {-1, 0, 0, 1, -1, -1, 1, 1};
int dy[8] = {0, -1, 1, 0, -1, 1, -1, 1};
using Graph = vector<vector<int>>;//隣接リスト:G[i]にはiと隣接する頂点が入るよ。
using P = pair<int, int>;//BFSで利用。queueに入れる。
using lP = pair<ll, ll>;
using PP = pair<int, P>; //dijkstraで利用,priority_queueに入れる。
using p_queue = priority_queue<int, vector<int>, greater<int>>;//小さい順
//struct edge{int to, cost};
int max_N = 100000;

//      Union Find木の実装
int par[100000];
int rnk[100000];

//n個のnodeで初期化する関数
void init(int n) {
    REP(i, n) {
        par[i] = i;
        rnk[i] = 0;
    }
}

//木の根を求める。この際，親は根に更新される。
int find(int x) {
    if (par[x] == x) { return x; }
    else { return par[x] = find(par[x]); }
}

//xとyの属する集合を融合
void unite(int x, int y) {
    x = find(x);
    y = find(y);
    if (x == y) { return; }

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

//xとyが同じ集合に属するか判定
bool same(int x, int y) {
    return find(x) == find(y);
}


//番号ズレ注意！！
int main() {
    int N, Q;
    cin >> N >> Q;
    init(N);
    REP(i, Q) {
        int P, A, B;
        cin >> P >> A >> B;
        if (P == 0) {
            unite(A, B);
        }
        if (P == 1) {
            YesNo(same(A, B));
        }
    }

}

Submission Info

Judge Result