psVAUv - Online IDE & Debugging Tool

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

int n, q;
const int N = 1e5 + 5;
int parent[N];

void make_set(){
      // Đặt nút "cha" của mỗi tập hợp là chính nó
      for (int i = 1; i <= n; i++) parent[i] = i;
}

int find(int x){
      // Ta sử dụng kĩ thuật nén đường đi để tối ưu
      if (parent[x] == x) return x;
      int p = find(parent[x]);
      parent[x] = p;
      return p;
}

void set_union(int a, int b){
      // Đặt nút "cha" của b là a
      a = find(a); b = find(b);
      if (a != b){
            parent[b] = a;
      }
}

void input(){
      cin >> n >> q;
      make_set();
      for (int i = 1; i <= q; i++){
            int t, u, v;
            cin >> t >> u >> v;

            if (t == 1){
                  // Nối 2 đỉnh u, v
                  set_union(u, v);
            }
            else{
                  // Kiểm tra xem 2 đỉnh u, v có được kết nối hay không?
                  if (find(u) == find(v)) cout << "YES" << endl;
                  else cout << "NO" << endl;
            }
      }
}

int main(){
      input();

      return 0;
}