Problem p03787 (C++)

Buggy submission (status: Runtime Error):
#include <algorithm>
#include <cassert>
#include <cmath>
#include <cstdio>
#include <fstream>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <sstream>
#include <string>
#include <vector>
using namespace std;

#define REP(i, m, n) for (int i = (m); i < (int)(n); i++)
#define RREP(i, m, n) for (int i = (int)(n - 1); i >= m; i--)
#define rep(i, n) REP(i, 0, n)
#define rrep(i, n) RREP(i, 0, n)
#define all(a) (a).begin(), (a).end()
#define rall(a) (a).rbegin(), (a).rend()
#define aut(r, v) __typeof(v) r = (v)
#define each(it, o) for (aut(it, (o).begin()); it != (o).end(); ++it)
#define reach(it, o) for (aut(it, (o).rbegin()); it != (o).rend(); ++it)
#define fi first
#define se second
#define debug(...)                                                             \
  {                                                                            \
    cerr << "[L" << __LINE__ << "] ";                                          \
    _debug(__VA_ARGS__);                                                       \
  }

template <typename T1, typename T2>
ostream &operator<<(ostream &o, const pair<T1, T2> &p) {
  return o << "(" << p.first << ", " << p.second << ")";
}
template <typename T> string join(const vector<T> &v, string del = ", ") {
  stringstream s;
  rep(i, v.size()) s << del << v[i];
  return s.str().substr(del.size());
}
template <typename T> ostream &operator<<(ostream &o, const vector<T> &v) {
  if (v.size())
    o << "[" << join(v) << "]";
  return o;
}
template <typename T>
ostream &operator<<(ostream &o, const vector<vector<T>> &vv) {
  int l = vv.size();
  if (l) {
    o << endl;
    rep(i, l) {
      o << (i == 0 ? "[ " : ",\n  ") << vv[i] << (i == l - 1 ? " ]" : "");
    }
  }
  return o;
}
template <typename T> ostream &operator<<(ostream &o, const set<T> &st) {
  vector<T> v(st.begin(), st.end());
  o << "{ " << join(v) << " }";
  return o;
}
template <typename T1, typename T2>
ostream &operator<<(ostream &o, const map<T1, T2> &m) {
  each(p, m) {
    o << (p == m.begin() ? "{ " : ",\n  ") << *p
      << (p == --m.end() ? " }" : "");
  }
  return o;
}
inline void _debug() { cerr << endl; }
template <class First, class... Rest>
void _debug(const First &first, const Rest &...rest) {
  cerr << first << " ";
  _debug(rest...);
}

typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vl;
typedef vector<vl> vvl;

const double PI = (1 * acos(0.0));
const double INF = 0x3f3f3f3f;
const double INFL = 0x3f3f3f3f3f3f3f3fLL;
const double EPS = 1e-9;
const double mod = 1e9 + 7;

inline void finput(string filename) { freopen(filename.c_str(), "r", stdin); }

vi g[100010];
int c[100010];
int bt;

void dfs(int v) {
  each(u, g[v]) {
    if (c[*u] == -1) {
      c[*u] = c[v] ^ 1;
      dfs(*u);
    } else {
      if ((c[*u] ^ c[v]) != 1)
        bt = 0;
    }
  }
}

int main() {
  ios_base::sync_with_stdio(0);
  finput("./input");
  int n, m;
  cin >> n >> m;
  rep(i, m) {
    int u, v;
    cin >> u >> v;
    u--;
    v--;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  fill(c, c + n, -1);

  ll a[3] = {0};
  rep(i, n) {
    if (c[i] != -1)
      continue;
    if (g[i].size() == 0) {
      c[i] = 0;
      a[0]++;
    } else {
      bt = 1;
      c[i] = 0;
      dfs(i);
      if (bt)
        a[1]++;
      else
        a[2]++;
    }
  }

  ll ans = 0;
  REP(i, 1, 3)
  REP(j, 1, 3)
  if (i == 1 && j == 1)
    ans += 2ll * a[i] * a[j];
  else
    ans += a[i] * a[j];

  ans += 2ll * a[0] * (n - a[0]) + a[0] * a[0];
  cout << ans << endl;

  return 0;
}


Fixed submission:
#include <algorithm>
#include <cassert>
#include <cmath>
#include <cstdio>
#include <fstream>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <sstream>
#include <string>
#include <vector>
using namespace std;

#define REP(i, m, n) for (int i = (m); i < (int)(n); i++)
#define RREP(i, m, n) for (int i = (int)(n - 1); i >= m; i--)
#define rep(i, n) REP(i, 0, n)
#define rrep(i, n) RREP(i, 0, n)
#define all(a) (a).begin(), (a).end()
#define rall(a) (a).rbegin(), (a).rend()
#define aut(r, v) __typeof(v) r = (v)
#define each(it, o) for (aut(it, (o).begin()); it != (o).end(); ++it)
#define reach(it, o) for (aut(it, (o).rbegin()); it != (o).rend(); ++it)
#define fi first
#define se second
#define debug(...)                                                             \
  {                                                                            \
    cerr << "[L" << __LINE__ << "] ";                                          \
    _debug(__VA_ARGS__);                                                       \
  }

template <typename T1, typename T2>
ostream &operator<<(ostream &o, const pair<T1, T2> &p) {
  return o << "(" << p.first << ", " << p.second << ")";
}
template <typename T> string join(const vector<T> &v, string del = ", ") {
  stringstream s;
  rep(i, v.size()) s << del << v[i];
  return s.str().substr(del.size());
}
template <typename T> ostream &operator<<(ostream &o, const vector<T> &v) {
  if (v.size())
    o << "[" << join(v) << "]";
  return o;
}
template <typename T>
ostream &operator<<(ostream &o, const vector<vector<T>> &vv) {
  int l = vv.size();
  if (l) {
    o << endl;
    rep(i, l) {
      o << (i == 0 ? "[ " : ",\n  ") << vv[i] << (i == l - 1 ? " ]" : "");
    }
  }
  return o;
}
template <typename T> ostream &operator<<(ostream &o, const set<T> &st) {
  vector<T> v(st.begin(), st.end());
  o << "{ " << join(v) << " }";
  return o;
}
template <typename T1, typename T2>
ostream &operator<<(ostream &o, const map<T1, T2> &m) {
  each(p, m) {
    o << (p == m.begin() ? "{ " : ",\n  ") << *p
      << (p == --m.end() ? " }" : "");
  }
  return o;
}
inline void _debug() { cerr << endl; }
template <class First, class... Rest>
void _debug(const First &first, const Rest &...rest) {
  cerr << first << " ";
  _debug(rest...);
}

typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vl;
typedef vector<vl> vvl;

const double PI = (1 * acos(0.0));
const double INF = 0x3f3f3f3f;
const double INFL = 0x3f3f3f3f3f3f3f3fLL;
const double EPS = 1e-9;
const double mod = 1e9 + 7;

inline void finput(string filename) { freopen(filename.c_str(), "r", stdin); }

vi g[100010];
int c[100010];
int bt;

void dfs(int v) {
  each(u, g[v]) {
    if (c[*u] == -1) {
      c[*u] = c[v] ^ 1;
      dfs(*u);
    } else {
      if ((c[*u] ^ c[v]) != 1)
        bt = 0;
    }
  }
}

int main() {
  ios_base::sync_with_stdio(0);
  // finput("./input");
  int n, m;
  cin >> n >> m;
  rep(i, m) {
    int u, v;
    cin >> u >> v;
    u--;
    v--;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  fill(c, c + n, -1);

  ll a[3] = {0};
  rep(i, n) {
    if (c[i] != -1)
      continue;
    if (g[i].size() == 0) {
      c[i] = 0;
      a[0]++;
    } else {
      bt = 1;
      c[i] = 0;
      dfs(i);
      if (bt)
        a[1]++;
      else
        a[2]++;
    }
  }

  ll ans = 0;
  REP(i, 1, 3)
  REP(j, 1, 3)
  if (i == 1 && j == 1)
    ans += 2ll * a[i] * a[j];
  else
    ans += a[i] * a[j];

  ans += 2ll * a[0] * (n - a[0]) + a[0] * a[0];
  cout << ans << endl;

  return 0;
}