Problem p02558 (C++)

Buggy submission (status: Runtime Error):
#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx")
#include <bits/stdc++.h>
#define _GLIBCXX_DEBUG
using namespace std;
using ll = long long;
using vec = vector<ll>;
using vect = vector<double>;
using Graph = vector<vector<ll>>;
#define endl '\n'
#define loop(i, n) for (int i = 0; i < n; i++)
#define Loop(i, m, n) for (ll i = m; i < n; i++)
#define pool(i, n) for (ll i = n; i >= 0; i--)
#define Pool(i, m, n) for (ll i = n; i >= m; i--)
#define modd 1000000007ll
// #define modd 998244353ll
#define flagcount(bit) __builtin_popcount(bit)
#define flag(x) (1ll << x)
#define flagadd(bit, x) bit |= flag(x)
#define flagpop(bit, x) bit &= ~flag(x)
#define flagon(bit, i) bit &flag(i)
#define flagoff(bit, i) !(bit & (1ll << i))
#define all(v) v.begin(), v.end()
#define low2way(v, x) lower_bound(all(v), x)
#define high2way(v, x) upper_bound(all(v), x)
#define idx_lower(v, x)                                                        \
  (distance(v.begin(), low2way(v, x))) // 配列vでx未満の要素数を返す
#define idx_upper(v, x)                                                        \
  (distance(v.begin(), high2way(v, x))) // 配列vでx以下の要素数を返す
#define idx_lower2(v, x)                                                       \
  (v.size() - idx_lower(v, x)) // 配列vでx以上の要素数を返す
#define idx_upper2(v, x)                                                       \
  (v.size() - idx_upper(v, x)) // 配列vでxより大きい要素の数を返す
#define putout(a) cout << a << '\n'
#define Sum(v) accumulate(all(v), 0ll)
ll ctoi(char c) {
  if (c >= '0' && c <= '9') {
    return c - '0';
  }
  return -1;
}
template <typename T> string make_string(T N) {
  string ret;
  T now = N;
  while (now > 0) {
    T x = now % 10;
    ret += (char)('0' + x);
    now /= 10;
  }
  reverse(all(ret));
  return ret;
}
template <typename T> T gcd(T a, T b) {
  if (a % b == 0) {
    return (b);
  } else {
    return (gcd(b, a % b));
  }
}
template <typename T> T lcm(T x, T y) {
  T z = gcd(x, y);
  return x * y / z;
}
template <typename T> bool primejudge(T n) {
  if (n < 2)
    return false;
  else if (n == 2)
    return true;
  else if (n % 2 == 0)
    return false;
  double sqrtn = sqrt(n);
  for (T i = 3; i < sqrtn + 1; i++) {
    if (n % i == 0) {
      return false;
    }
    i++;
  }
  return true;
}
template <typename T> bool chmax(T &a, const T &b) {
  if (a < b) {
    a = b; // aをbで更新
    return true;
  }
  return false;
}
template <typename T> bool chmin(T &a, const T &b) {
  if (a > b) {
    a = b; // aをbで更新
    return true;
  }
  return false;
}
// 場合によって使い分ける
// const ll dx[4]={1,0,-1,0};
// const ll dy[4]={0,1,0,-1};
const ll dx[8] = {1, 1, 0, -1, -1, -1, 0, 1};
const ll dy[8] = {0, 1, 1, 1, 0, -1, -1, -1};
class FastIO {
  static const int rdata_sz = (1 << 20), wdata_sz = (1 << 20);
  char rdata[rdata_sz], wdata[wdata_sz], *rb, *wb;
  char tmp_s[20];

public:
  FastIO() {
    fread(rdata, 1, rdata_sz, stdin);
    rb = rdata;
    wb = wdata;
  }
  ~FastIO() { fwrite(wdata, 1, wb - wdata, stdout); }

  template <typename T> inline void read(T &x) {
    bool neg = false;
    x = 0;
    while ((*rb < '0' || *rb > '9') && *rb != '-')
      ++rb;
    if (*rb == '-') {
      neg = true;
      ++rb;
    }
    while ('0' <= *rb && *rb <= '9') {
      x = 10 * x + (*rb - '0');
      ++rb;
    }
    if (neg)
      x = -x;
  }

#define pc(x) *(wb++) = x
  template <typename T> inline void write(T x) {
    if (x == 0) {
      pc('0');
      pc('\n');
      return;
    }
    if (x < 0) {
      pc('-');
      x = -x;
    }
    char *t = tmp_s;
    while (x) {
      T y = x / 10;
      *(t++) = (x - y * 10) + '0';
      x = y;
    }
    while (t != tmp_s)
      pc(*(--t));
    pc('\n');
  }
#undef pc
};
struct union_find {
  vector<int> par;  // 親の番号　
  vector<int> rank; // 木の深さ(根のランクは0)
  vector<int> siz;  // 要素xが根なら木の頂点数を格納する
  // 初期化子リストを用いた初期化
  union_find(int N) : par(N), rank(N), siz(N) {
    for (int i = 0; i < N; i++) {
      par[i] = i;
      rank[i] = 0;
      siz[i] = 1;
    }
  }
  // 要素xが所属する木の根を再帰的に発見する
  int root(int x) {
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]); // 経路圧縮
  }
  // 要素xが属する木と要素yが属する木の併合
  void unite(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    if (rx == ry)
      return; // 同じ木に属してたらそのまま
    if (rank[rx] < rank[ry]) {
      par[rx] = ry; // 根がryの木に併合
      siz[ry] = siz[rx] + siz[ry];
    } else {
      par[ry] = rx; // 根がrxの木に併合
      siz[rx] = siz[rx] + siz[ry];
      if (rank[rx] == rank[ry])
        rank[rx]++;
    }
  }
  // 要素xが属する木と要素yが属する木が同じならtrueを返す
  bool same(int x, int y) { return root(x) == root(y); }
  // 要素xが属する木の頂点数を返す
  int size(int x) { return siz[root(x)]; }
};
int main() {
  FastIO io;
  int N, Q;
  io.read(N);
  io.read(Q);
  union_find tree(N);
  for (int i = 0; i < Q; i++) {
    int a, b, c;
    io.read(a);
    io.read(b);
    io.read(c);
    if (a == 1)
      if (tree.same(b, c))
        io.write(1);
      else
        io.write(0);
    if (a == 0)
      tree.unite(b, c);
  }
  return 0;
}


Fixed submission:
#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx")
#include <bits/stdc++.h>
#define _GLIBCXX_DEBUG
using namespace std;
using ll = long long;
using vec = vector<ll>;
using vect = vector<double>;
using Graph = vector<vector<ll>>;
#define endl '\n'
#define loop(i, n) for (int i = 0; i < n; i++)
#define Loop(i, m, n) for (ll i = m; i < n; i++)
#define pool(i, n) for (ll i = n; i >= 0; i--)
#define Pool(i, m, n) for (ll i = n; i >= m; i--)
#define modd 1000000007ll
// #define modd 998244353ll
#define flagcount(bit) __builtin_popcount(bit)
#define flag(x) (1ll << x)
#define flagadd(bit, x) bit |= flag(x)
#define flagpop(bit, x) bit &= ~flag(x)
#define flagon(bit, i) bit &flag(i)
#define flagoff(bit, i) !(bit & (1ll << i))
#define all(v) v.begin(), v.end()
#define low2way(v, x) lower_bound(all(v), x)
#define high2way(v, x) upper_bound(all(v), x)
#define idx_lower(v, x)                                                        \
  (distance(v.begin(), low2way(v, x))) // 配列vでx未満の要素数を返す
#define idx_upper(v, x)                                                        \
  (distance(v.begin(), high2way(v, x))) // 配列vでx以下の要素数を返す
#define idx_lower2(v, x)                                                       \
  (v.size() - idx_lower(v, x)) // 配列vでx以上の要素数を返す
#define idx_upper2(v, x)                                                       \
  (v.size() - idx_upper(v, x)) // 配列vでxより大きい要素の数を返す
#define putout(a) cout << a << '\n'
#define Sum(v) accumulate(all(v), 0ll)
ll ctoi(char c) {
  if (c >= '0' && c <= '9') {
    return c - '0';
  }
  return -1;
}
template <typename T> string make_string(T N) {
  string ret;
  T now = N;
  while (now > 0) {
    T x = now % 10;
    ret += (char)('0' + x);
    now /= 10;
  }
  reverse(all(ret));
  return ret;
}
template <typename T> T gcd(T a, T b) {
  if (a % b == 0) {
    return (b);
  } else {
    return (gcd(b, a % b));
  }
}
template <typename T> T lcm(T x, T y) {
  T z = gcd(x, y);
  return x * y / z;
}
template <typename T> bool primejudge(T n) {
  if (n < 2)
    return false;
  else if (n == 2)
    return true;
  else if (n % 2 == 0)
    return false;
  double sqrtn = sqrt(n);
  for (T i = 3; i < sqrtn + 1; i++) {
    if (n % i == 0) {
      return false;
    }
    i++;
  }
  return true;
}
template <typename T> bool chmax(T &a, const T &b) {
  if (a < b) {
    a = b; // aをbで更新
    return true;
  }
  return false;
}
template <typename T> bool chmin(T &a, const T &b) {
  if (a > b) {
    a = b; // aをbで更新
    return true;
  }
  return false;
}
// 場合によって使い分ける
// const ll dx[4]={1,0,-1,0};
// const ll dy[4]={0,1,0,-1};
const ll dx[8] = {1, 1, 0, -1, -1, -1, 0, 1};
const ll dy[8] = {0, 1, 1, 1, 0, -1, -1, -1};
class FastIO {
  static const int rdata_sz = (1 << 23), wdata_sz = (1 << 23);
  char rdata[rdata_sz], wdata[wdata_sz], *rb, *wb;
  char tmp_s[20];

public:
  FastIO() {
    fread(rdata, 1, rdata_sz, stdin);
    rb = rdata;
    wb = wdata;
  }
  ~FastIO() { fwrite(wdata, 1, wb - wdata, stdout); }

  template <typename T> inline void read(T &x) {
    bool neg = false;
    x = 0;
    while ((*rb < '0' || *rb > '9') && *rb != '-')
      ++rb;
    if (*rb == '-') {
      neg = true;
      ++rb;
    }
    while ('0' <= *rb && *rb <= '9') {
      x = 10 * x + (*rb - '0');
      ++rb;
    }
    if (neg)
      x = -x;
  }

#define pc(x) *(wb++) = x
  template <typename T> inline void write(T x) {
    if (x == 0) {
      pc('0');
      pc('\n');
      return;
    }
    if (x < 0) {
      pc('-');
      x = -x;
    }
    char *t = tmp_s;
    while (x) {
      T y = x / 10;
      *(t++) = (x - y * 10) + '0';
      x = y;
    }
    while (t != tmp_s)
      pc(*(--t));
    pc('\n');
  }
#undef pc
};
struct union_find {
  vector<int> par;  // 親の番号　
  vector<int> rank; // 木の深さ(根のランクは0)
  vector<int> siz;  // 要素xが根なら木の頂点数を格納する
  // 初期化子リストを用いた初期化
  union_find(int N) : par(N), rank(N), siz(N) {
    for (int i = 0; i < N; i++) {
      par[i] = i;
      rank[i] = 0;
      siz[i] = 1;
    }
  }
  // 要素xが所属する木の根を再帰的に発見する
  int root(int x) {
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]); // 経路圧縮
  }
  // 要素xが属する木と要素yが属する木の併合
  void unite(int x, int y) {
    int rx = root(x);
    int ry = root(y);
    if (rx == ry)
      return; // 同じ木に属してたらそのまま
    if (rank[rx] < rank[ry]) {
      par[rx] = ry; // 根がryの木に併合
      siz[ry] = siz[rx] + siz[ry];
    } else {
      par[ry] = rx; // 根がrxの木に併合
      siz[rx] = siz[rx] + siz[ry];
      if (rank[rx] == rank[ry])
        rank[rx]++;
    }
  }
  // 要素xが属する木と要素yが属する木が同じならtrueを返す
  bool same(int x, int y) { return root(x) == root(y); }
  // 要素xが属する木の頂点数を返す
  int size(int x) { return siz[root(x)]; }
};
int main() {
  FastIO io;
  int N, Q;
  io.read(N);
  io.read(Q);
  union_find tree(N);
  for (int i = 0; i < Q; i++) {
    int a, b, c;
    io.read(a);
    io.read(b);
    io.read(c);
    if (a == 1)
      if (tree.same(b, c))
        io.write(1);
      else
        io.write(0);
    if (a == 0)
      tree.unite(b, c);
  }
  return 0;
}