Problem p03108 (C++)

Buggy submission (status: Runtime Error):
#include <bits/stdc++.h>
#define rep(i, n) for (int(i) = 0; (i) < (n); (i)++)
#define all(x) (x).begin(), (x).end()

using namespace std;
using ll = long long;
using P = pair<int, int>;

const int INF = 1001001001;
const vector<int> di = {-1, 0, 1, 0};
const vector<int> dj = {0, -1, 0, 1};

void chmin(int &a, int b) {
  if (a > b)
    a = b;
}

ll GCD(ll a, ll b) {
  if (b == 0)
    return a;
  else
    return GCD(b, a % b);
}
ll LCM(ll a, ll b) { return a * b / GCD(a, b); }

struct UnionFind {
  // parent[a] = b: "aの親がbである"
  // 親の場合は-(その集合のサイズ)を返す。
  vector<int> parent;

  // 全て根であるとみなす。
  // parent[x] == -1のとき、そのノードは根である。
  UnionFind(int n = 0) : parent(n, -1) {}

  // xがどのグループに属しているかを返す。
  int find(int x) {
    if (parent[x] < 0)
      return x; // 今の頂点が根ならその値を返す
    return parent[x] = find(parent[x]); // 経路縮約（メモ化）
  }

  // AとBをつなげる。AとBを直接つなぐのではなく、root(A)とroot(B)をつなげる。
  bool unite(int x,
             int y) { // Minimum Spanning Treeのためにboolを返す実装にしている。
    x = find(x); // それぞれのノードの親を取得する。
    y = find(y);

    if (x == y)
      return false; // すでにくっついているのでくっつけない。
    // 大きい方に小さい方をくっつけたい。
    if (size(x) < size(y))
      swap(
          x,
          y); // マイナスサイズなので不等号が逆。xのサイズの方が大きいようにする。
    parent[x] += parent[y]; // 連結成分のサイズを管理するためのコード
    parent[y] = x; // xにyをくっつける。yの値をxの根とする。

    return true;
  }

  bool same(int x, int y) { return find(x) == find(y); }
  int size(int x) { return -parent[find(x)]; }
};

int main() {
  int n, m;
  cin >> n >> m;

  vector<int> a(n), b(m);
  rep(i, m) {
    cin >> a[i] >> b[i];
    a[i]--;
    b[i]--;
  }

  vector<ll> ans(m);
  ans[m - 1] = (ll)n * (n - 1) / 2;

  UnionFind uni(n); // 頂点数を渡す。

  for (int i = m - 1; i >= 1; --i) {
    ans[i - 1] = ans[i];

    // つながっていなかったものがつながったとき
    if (uni.find(a[i]) != uni.find(b[i])) {
      ans[i - 1] -= (ll)uni.size(a[i]) * uni.size(b[i]);
      uni.unite(a[i], b[i]);
    }
  }

  rep(i, m) { cout << ans[i] << endl; }
}

Fixed submission:
#include <bits/stdc++.h>
#define rep(i, n) for (int(i) = 0; (i) < (n); (i)++)
#define all(x) (x).begin(), (x).end()

using namespace std;
using ll = long long;
using P = pair<int, int>;

const int INF = 1001001001;
const vector<int> di = {-1, 0, 1, 0};
const vector<int> dj = {0, -1, 0, 1};

void chmin(int &a, int b) {
  if (a > b)
    a = b;
}

ll GCD(ll a, ll b) {
  if (b == 0)
    return a;
  else
    return GCD(b, a % b);
}
ll LCM(ll a, ll b) { return a * b / GCD(a, b); }

struct UnionFind {
  // parent[a] = b: "aの親がbである"
  // 親の場合は-(その集合のサイズ)を返す。
  vector<int> parent;

  // 全て根であるとみなす。
  // parent[x] == -1のとき、そのノードは根である。
  UnionFind(int n = 0) : parent(n, -1) {}

  // xがどのグループに属しているかを返す。
  int find(int x) {
    if (parent[x] < 0)
      return x; // 今の頂点が根ならその値を返す
    return parent[x] = find(parent[x]); // 経路縮約（メモ化）
  }

  // AとBをつなげる。AとBを直接つなぐのではなく、root(A)とroot(B)をつなげる。
  bool unite(int x,
             int y) { // Minimum Spanning Treeのためにboolを返す実装にしている。
    x = find(x); // それぞれのノードの親を取得する。
    y = find(y);

    if (x == y)
      return false; // すでにくっついているのでくっつけない。
    // 大きい方に小さい方をくっつけたい。
    if (size(x) < size(y))
      swap(
          x,
          y); // マイナスサイズなので不等号が逆。xのサイズの方が大きいようにする。
    parent[x] += parent[y]; // 連結成分のサイズを管理するためのコード
    parent[y] = x; // xにyをくっつける。yの値をxの根とする。

    return true;
  }

  bool same(int x, int y) { return find(x) == find(y); }
  int size(int x) { return -parent[find(x)]; }
};

int main() {
  int n, m;
  cin >> n >> m;

  vector<int> a(m), b(m);
  rep(i, m) {
    cin >> a[i] >> b[i];
    a[i]--;
    b[i]--;
  }

  vector<ll> ans(m);
  ans[m - 1] = (ll)n * (n - 1) / 2;

  UnionFind uni(n); // 頂点数を渡す。

  for (int i = m - 1; i >= 1; --i) {
    ans[i - 1] = ans[i];

    // つながっていなかったものがつながったとき
    if (uni.find(a[i]) != uni.find(b[i])) {
      ans[i - 1] -= (ll)uni.size(a[i]) * uni.size(b[i]);
      uni.unite(a[i], b[i]);
    }
  }

  rep(i, m) { cout << ans[i] << endl; }
}