Problem p02343 (C++)

Buggy submission (status: Runtime Error):
#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <functional>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <vector>
using namespace std;

#define Rep(b, e, i) for (int i = b; i <= e; i++)
#define Repr(e, b, i) for (int i = e; i >= b; i--)
#define rep(n, i) Rep(0, n - 1, i)
#define repr(n, i) Repr(n - 1, 0, i)
#define all(v) (v).begin(), (v).end()
#define pb(v) push_back(v)
#define uniq(v) (v).erase(unique(all(v)), (v).end())
#define bitcnt(x) __builtin_popcount(x)
#define fst first
#define snd second
#define Pqaz(T) priority_queue<T, vector<T>, greater<T>>
#define Pqza(T) priority_queue<T>
#define ENJYU                                                                  \
  std::ios::sync_with_stdio(false);                                            \
  std::cin.tie(0);

struct UnionFind {
  // number of parents(par[i] = i => root)
  vector<int> par;
  // rank of tree
  vector<int> rank;

  UnionFind(int n) : par(n), rank(n, 1) {
    // initialize
    rep(n, i) par[i] = i;
  }

  // find root (keiroasshuku at the same time)
  int root(int x) {
    if (par[x] == x)
      return x;
    else
      return par[x] = root(par[x]);
  }

  // unite
  void unite(int x, int y) {
    x = root(x);
    y = root(y);
    if (x == y)
      return;
    if (rank[x] > rank[y])
      swap(x, y);
    par[x] = y;
    if (rank[x] == rank[y])
      rank[y]++;
  }

  // same tree?
  bool same(int x, int y) { return root(x) == root(y); }
};

void solve(void) {
  int N, Q;
  cin >> N >> Q;
  UnionFind uf(N);
  while (Q--) {
    int c, x, y;
    cin >> c >> x >> y;
    x--, y--;
    if (c)
      cout << uf.same(x, y) << endl;
    else
      uf.unite(x, y);
  }
}

int main(void) {
  solve();
  // cout << "yui(*-v・)yui" << endl;
  return 0;
}


Fixed submission:
#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <functional>
#include <iostream>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <vector>
using namespace std;

#define Rep(b, e, i) for (int i = b; i <= e; i++)
#define Repr(e, b, i) for (int i = e; i >= b; i--)
#define rep(n, i) Rep(0, n - 1, i)
#define repr(n, i) Repr(n - 1, 0, i)
#define all(v) (v).begin(), (v).end()
#define pb(v) push_back(v)
#define uniq(v) (v).erase(unique(all(v)), (v).end())
#define bitcnt(x) __builtin_popcount(x)
#define fst first
#define snd second
#define Pqaz(T) priority_queue<T, vector<T>, greater<T>>
#define Pqza(T) priority_queue<T>
#define ENJYU                                                                  \
  std::ios::sync_with_stdio(false);                                            \
  std::cin.tie(0);

struct UnionFind {
  // number of parents(par[i] = i => root)
  vector<int> par;
  // rank of tree
  vector<int> rank;

  UnionFind(int n) : par(n), rank(n, 1) {
    // initialize
    rep(n, i) par[i] = i;
  }

  // find root (keiroasshuku at the same time)
  int root(int x) {
    if (par[x] == x)
      return x;
    else
      return par[x] = root(par[x]);
  }

  // unite
  void unite(int x, int y) {
    x = root(x);
    y = root(y);
    if (x == y)
      return;
    if (rank[x] > rank[y])
      swap(x, y);
    par[x] = y;
    if (rank[x] == rank[y])
      rank[y]++;
  }

  // same tree?
  bool same(int x, int y) { return root(x) == root(y); }
};

void solve(void) {
  int N, Q;
  cin >> N >> Q;
  UnionFind uf(N);
  while (Q--) {
    int c, x, y;
    cin >> c >> x >> y;
    if (c)
      cout << uf.same(x, y) << endl;
    else
      uf.unite(x, y);
  }
}

int main(void) {
  solve();
  // cout << "yui(*-v・)yui" << endl;
  return 0;
}