Problem p03363 (C++)

Buggy submission (status: Runtime Error):
#include <algorithm>
#include <bits/stdc++.h>
#include <bitset>
#include <cmath>
#include <cstring>
#include <deque>
#include <iomanip>
#include <iostream>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;

typedef long long ll;
#define rep(i, n) for (ll i = 0; i < n; ++i)
#define exout(x) printf("%.10f\n", x)
const double pi = acos(-1.0);
const ll MOD = 1000000007;
const ll INF = 1e10;
const ll MAX_N = 201010;

// 組み合わせの余りを求める
ll fac[MAX_N], finv[MAX_N], inv[MAX_N];
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX_N; i++) {
    fac[i] = fac[i - 1] * i;
  }
}

// 二項係数計算
long long COM(ll n, ll k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] / (fac[k] * fac[n - k]);
}

ll dx[4] = {0, 0, -1, 1};
ll dy[4] = {-1, 1, 0, 0};
char c[310][310];
ll dp[101];

// long longしか使わない
// 素数は1より大きい
// lower_boundは指定したkey以上の要素の一番左のイテレータをかえす
// upper_boundは指定したkeyより大きい要素の一番左のイテレータをかえす
int main() {
  ll n;
  cin >> n;
  vector<ll> a(n);
  vector<ll> sum(n);
  rep(i, n) {
    cin >> a[i];
    sum[i] = a[i];
  }
  rep(i, n - 1) { sum[i + 1] += sum[i]; }
  map<ll, ll> mp;
  ll ans = 0;
  rep(i, n) {
    mp[sum[i]]++;
    if (sum[i] == 0)
      ans++;
  }
  COMinit();
  for (auto p : mp) {
    ll x = p.second;
    if (x > 1) {
      ans += COM(x, 2);
    }
  }
  cout << ans << endl;
  return 0;
}


Fixed submission:
#include <algorithm>
#include <bits/stdc++.h>
#include <bitset>
#include <cmath>
#include <cstring>
#include <deque>
#include <iomanip>
#include <iostream>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;

typedef long long ll;
#define rep(i, n) for (ll i = 0; i < n; ++i)
#define exout(x) printf("%.10f\n", x)
const double pi = acos(-1.0);
const ll MOD = 1000000007;
const ll INF = 1e10;
const ll MAX_N = 201010;

// 組み合わせの余りを求める
ll fac[MAX_N], finv[MAX_N], inv[MAX_N];
void COMinit() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (int i = 2; i < MAX_N; i++) {
    fac[i] = fac[i - 1] * i;
  }
}

// 二項係数計算
long long COM(ll n, ll k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] / (fac[k] * fac[n - k]);
}

ll dx[4] = {0, 0, -1, 1};
ll dy[4] = {-1, 1, 0, 0};
char c[310][310];
ll dp[101];

// long longしか使わない
// 素数は1より大きい
// lower_boundは指定したkey以上の要素の一番左のイテレータをかえす
// upper_boundは指定したkeyより大きい要素の一番左のイテレータをかえす
int main() {
  ll n;
  cin >> n;
  vector<ll> a(n);
  vector<ll> sum(n);
  rep(i, n) {
    cin >> a[i];
    sum[i] = a[i];
  }
  rep(i, n - 1) { sum[i + 1] += sum[i]; }
  map<ll, ll> mp;
  ll ans = 0;
  rep(i, n) {
    mp[sum[i]]++;
    if (sum[i] == 0)
      ans++;
  }
  COMinit();
  for (auto p : mp) {
    ll x = p.second;
    if (x > 1) {
      ans += x * (x - 1) / 2;
    }
  }
  cout << ans << endl;
  return 0;
}