Problem p03045 (C++)

Buggy submission (status: Runtime Error):
#include <bits/stdc++.h>

#define _overload(_1, _2, _3, name, ...) name
#define _rep(i, n) _range(i, 0, n)
#define _range(i, a, b) for (int i = int(a); i < int(b); ++i)
#define rep(...) _overload(__VA_ARGS__, _range, _rep, )(__VA_ARGS__)

#define _rrep(i, n) _rrange(i, n, 0)
#define _rrange(i, a, b) for (int i = int(a) - 1; i >= int(b); --i)
#define rrep(...) _overload(__VA_ARGS__, _rrange, _rrep, )(__VA_ARGS__)

#define _all(arg) begin(arg), end(arg)
#define uniq(arg) sort(_all(arg)), (arg).erase(unique(_all(arg)), end(arg))
#define getidx(ary, key) lower_bound(_all(ary), key) - begin(ary)
#define clr(a, b) memset((a), (b), sizeof(a))
#define bit(n) (1LL << (n))
#define popcount(n) (__builtin_popcountll(n))

using namespace std;

template <class T> bool chmax(T &a, const T &b) {
  return (a < b) ? (a = b, 1) : 0;
}
template <class T> bool chmin(T &a, const T &b) {
  return (b < a) ? (a = b, 1) : 0;
}

typedef long long ll;
typedef long double R;
const R EPS = 1e-9L; // [-1000,1000]->EPS=1e-8 [-10000,10000]->EPS=1e-7
inline int sgn(const R &r) { return (r > EPS) - (r < -EPS); }
inline R sq(R x) { return sqrt(max(x, 0.0L)); }

const int dx[8] = {1, 0, -1, 0, 1, -1, -1, 1};
const int dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};

class UnionFind {
public:
  // 親の番号を格納する。親だった場合は-(その集合のサイズ)
  vector<int> Parent;

  // 作る時はParentの値を全て-1にする
  // こうすると全てバラバラが表現できる。
  UnionFind(int N) { Parent = vector<int>(N, -1); }

  // Aがどのグループに属しているか調べる
  int root(int A) {
    if (Parent[A] < 0)
      return A;
    return Parent[A] = root(Parent[A]);
  }

  int size(int A) {
    return -Parent[root(A)]; // 親を取ってきたい
  }

  bool connect(int A, int B) {
    // AとBを直接繋ぐのではなく、root(A)をroot(B)をくっつける
    A = root(A);
    B = root(B);
    if (A == B) {
      // すでにくっついているからくっつけない
      return false;
    }

    // 大きい方に小さい方をくっつけたい
    // 代償が逆だったらひっくり返す
    if (size(A) < size(B))
      swap(A, B);
    // Aのsizeを更新
    Parent[A] += Parent[B];
    // Bの親をAに変更する
    Parent[B] = A;

    return true;
  }
};

int main() {
  int N, M;
  cin >> N >> M;
  vector<int> x(M), y(M), z(M);
  long long ans = N;
  for (int i = 0; i < M; ++i) {
    cin >> x[i] >> y[i] >> z[i];
  }
  UnionFind Uni(N);
  for (int i = 0; i < M; ++i) {
    if (Uni.root(x[i]) != Uni.root(y[i])) {
      Uni.connect(x[i], y[i]);
      ans--;
    }
  }

  cout << ans << endl;

  return 0;
}


Fixed submission:
#include <bits/stdc++.h>

#define _overload(_1, _2, _3, name, ...) name
#define _rep(i, n) _range(i, 0, n)
#define _range(i, a, b) for (int i = int(a); i < int(b); ++i)
#define rep(...) _overload(__VA_ARGS__, _range, _rep, )(__VA_ARGS__)

#define _rrep(i, n) _rrange(i, n, 0)
#define _rrange(i, a, b) for (int i = int(a) - 1; i >= int(b); --i)
#define rrep(...) _overload(__VA_ARGS__, _rrange, _rrep, )(__VA_ARGS__)

#define _all(arg) begin(arg), end(arg)
#define uniq(arg) sort(_all(arg)), (arg).erase(unique(_all(arg)), end(arg))
#define getidx(ary, key) lower_bound(_all(ary), key) - begin(ary)
#define clr(a, b) memset((a), (b), sizeof(a))
#define bit(n) (1LL << (n))
#define popcount(n) (__builtin_popcountll(n))

using namespace std;

template <class T> bool chmax(T &a, const T &b) {
  return (a < b) ? (a = b, 1) : 0;
}
template <class T> bool chmin(T &a, const T &b) {
  return (b < a) ? (a = b, 1) : 0;
}

typedef long long ll;
typedef long double R;
const R EPS = 1e-9L; // [-1000,1000]->EPS=1e-8 [-10000,10000]->EPS=1e-7
inline int sgn(const R &r) { return (r > EPS) - (r < -EPS); }
inline R sq(R x) { return sqrt(max(x, 0.0L)); }

const int dx[8] = {1, 0, -1, 0, 1, -1, -1, 1};
const int dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};

class UnionFind {
public:
  // 親の番号を格納する。親だった場合は-(その集合のサイズ)
  vector<int> Parent;

  // 作る時はParentの値を全て-1にする
  // こうすると全てバラバラが表現できる。
  UnionFind(int N) { Parent = vector<int>(N, -1); }

  // Aがどのグループに属しているか調べる
  int root(int A) {
    if (Parent[A] < 0)
      return A;
    return Parent[A] = root(Parent[A]);
  }

  int size(int A) {
    return -Parent[root(A)]; // 親を取ってきたい
  }

  bool connect(int A, int B) {
    // AとBを直接繋ぐのではなく、root(A)をroot(B)をくっつける
    A = root(A);
    B = root(B);
    if (A == B) {
      // すでにくっついているからくっつけない
      return false;
    }

    // 大きい方に小さい方をくっつけたい
    // 代償が逆だったらひっくり返す
    if (size(A) < size(B))
      swap(A, B);
    // Aのsizeを更新
    Parent[A] += Parent[B];
    // Bの親をAに変更する
    Parent[B] = A;

    return true;
  }
};

int main() {
  int N, M;
  cin >> N >> M;
  vector<int> x(M), y(M), z(M);
  long long ans = N;
  for (int i = 0; i < M; ++i) {
    cin >> x[i] >> y[i] >> z[i];
    x[i]--, y[i]--, z[i]--;
  }
  UnionFind Uni(N);
  for (int i = 0; i < M; ++i) {
    if (Uni.root(x[i]) != Uni.root(y[i])) {
      Uni.connect(x[i], y[i]);
      ans--;
    }
  }

  cout << ans << endl;

  return 0;
}