Problem p03703 (C++)

Buggy submission (status: Runtime Error):
#define _USE_MATH_DEFINES
#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <fstream>
#include <iostream>
#include <limits>
#include <map>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <utility>
#include <vector>
using namespace std;
typedef long double ld;
typedef long long ll;
typedef vector<int> vint;
typedef vector<ll> vll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
typedef complex<ld> compd;
#define quickIO()                                                              \
  {                                                                            \
    cin.tie(0);                                                                \
    cout.sync_with_stdio(false);                                               \
  }
#define reach(i, a) for (auto i : a)
#define rep(i, n) for (int i = 0; i < (int)n; i++)
#define REP(i, n) for (int i = 0; i <= (int)n; i++)
#define srep(i, a, n) for (int i = a; i < (int)n; i++)
#define SREP(i, a, n) for (int i = a; i <= (int)n; i++)
#define rrep(i, n) for (int i = n - 1; i >= 0; i--)
#define RREP(i, n) for (int i = n; i >= 0; i--)
#define all(a) (a).begin(), (a).end()
#define mp(a, b) make_pair(a, b)
#define mt make_tuple
#define pb push_back
#define fst first
#define scn second
int bitcnt(ll x) {
  x = ((x & 0xAAAAAAAAAAAAAAAA) >> 1) + (x & 0x5555555555555555);
  x = ((x & 0xCCCCCCCCCCCCCCCC) >> 2) + (x & 0x3333333333333333);
  x = ((x & 0xF0F0F0F0F0F0F0F0) >> 4) + (x & 0x0F0F0F0F0F0F0F0F);
  x = ((x & 0xFF00FF00FF00FF00) >> 8) + (x & 0x00FF00FF00FF00FF);
  x = ((x & 0xFFFF0000FFFF0000) >> 16) + (x & 0x0000FFFF0000FFFF);
  x = ((x & 0xFFFFFFFF00000000) >> 32) + (x & 0x00000000FFFFFFFF);
  return x;
}
int bitcnt(int x) {
  x = ((x & 0xAAAAAAAA) >> 1) + (x & 0x55555555);
  x = ((x & 0xCCCCCCCC) >> 2) + (x & 0x33333333);
  x = ((x & 0xF0F0F0F0) >> 4) + (x & 0x0F0F0F0F);
  x = ((x & 0xFF00FF00) >> 8) + (x & 0x00FF00FF);
  x = ((x & 0xFFFF0000) >> 16) + (x & 0x0000FFFF);
  return x;
}
ll sqrt(ll x) {
  ll left = 0, right = x;
  rep(i, 100) {
    ll mid = (left + right) >> 1;
    if (mid * mid <= x)
      left = mid;
    else
      right = mid;
  }
  return left;
}
ll gcd(ll a, ll b) { return a % b == 0 ? b : gcd(b, a % b); }
#define debug(x) printf("Case #%d: ", x)
#define DEBUG 0
const ll inf = 1e18;
const ll mod = 1e9 + 7;
const ld eps = 1e-15;
const int dx[] = {1, 0, -1, 0, 0};
const int dy[] = {0, 1, 0, -1, 0};

// 取り敢えず平均がK→各要素Kずつ引いて、合計が非負ならOKと見ます。
// そうすると、累積和を取っておくと区間和が各区間についてO(1)、全体でO(N)で計算できます。ただ、このままでは間に合いません。
// ここで、累積和をsum[i]という配列に入れたとしたとき、「区間和の右側はsum[i]以下であれば全部カウントしてよい」という特徴があります。
// となると、左から見ていき、今まで出てきたsumを記録しておいて、新しいsum[i]に対してsum[i]が何番目に相当するかを調べることが出来れば解けそうです。
// 具体的には、記録した見ている箇所より左の集合の中で、「sum[i]より小さなもの」の個数がそこを左端とする個数です。（右側を探しているので、引くことに注意）
// 私はこれをBitIndexedTree(多分分科会であとでやります）を用いて実装しました。vectorは挿入に実は最悪O(N)かかるので、結構大変なので。
// 他の実装方法もあると思うので、頑張ってみてください。

// Bit Indexed Tree
// 0は使わないので注意。
int bit[200010];
int N;

void init(int n) {
  N = 1;
  while (N <= n)
    N <<= 1;
}

void update(int k, int a) {
  while (k < N) {
    bit[k] += a;
    k += k & -k;
  }
}

int getsum(int k) {
  int ret = 0;
  while (k > 0) {
    ret += bit[k];
    k -= k & -k;
  }
  return ret;
}

// 当然ですが、答えはint型に収まらないことがあるので注意

int main() {
  int n, k;
  cin >> n >> k;
  vector<ll> a(n);
  rep(i, n) {
    cin >> a[i];
    a[i] -= k;
  }
  vector<ll> sum(n + 1);
  SREP(i, 1, n) { sum[i] = sum[i - 1] + a[i - 1]; }
  // bit用にソートした累積和配列を用意
  // aの値そのままでbitをやると死ぬので、座標圧縮をします。
  vector<ll> sorted = sum;
  sort(all(sorted));

  init(n + 1);
  ll ret = 0;
  REP(i, n) {
    // sum[i]の位置を調べます。
    int addidx = lower_bound(all(sorted), sum[i]) - sorted.begin();
    // そこより左の個数をカウント（自身は含めません）
    ret += getsum(addidx + 1);
    // sum[i]を追加します。
    update(addidx + 1, 1);
  }
  cout << ret << endl;
  return 0;
}

Fixed submission:
#define _USE_MATH_DEFINES
#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <fstream>
#include <iostream>
#include <limits>
#include <map>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <utility>
#include <vector>
using namespace std;
typedef long double ld;
typedef long long ll;
typedef vector<int> vint;
typedef vector<ll> vll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
typedef complex<ld> compd;
#define quickIO()                                                              \
  {                                                                            \
    cin.tie(0);                                                                \
    cout.sync_with_stdio(false);                                               \
  }
#define reach(i, a) for (auto i : a)
#define rep(i, n) for (int i = 0; i < (int)n; i++)
#define REP(i, n) for (int i = 0; i <= (int)n; i++)
#define srep(i, a, n) for (int i = a; i < (int)n; i++)
#define SREP(i, a, n) for (int i = a; i <= (int)n; i++)
#define rrep(i, n) for (int i = n - 1; i >= 0; i--)
#define RREP(i, n) for (int i = n; i >= 0; i--)
#define all(a) (a).begin(), (a).end()
#define mp(a, b) make_pair(a, b)
#define mt make_tuple
#define pb push_back
#define fst first
#define scn second
int bitcnt(ll x) {
  x = ((x & 0xAAAAAAAAAAAAAAAA) >> 1) + (x & 0x5555555555555555);
  x = ((x & 0xCCCCCCCCCCCCCCCC) >> 2) + (x & 0x3333333333333333);
  x = ((x & 0xF0F0F0F0F0F0F0F0) >> 4) + (x & 0x0F0F0F0F0F0F0F0F);
  x = ((x & 0xFF00FF00FF00FF00) >> 8) + (x & 0x00FF00FF00FF00FF);
  x = ((x & 0xFFFF0000FFFF0000) >> 16) + (x & 0x0000FFFF0000FFFF);
  x = ((x & 0xFFFFFFFF00000000) >> 32) + (x & 0x00000000FFFFFFFF);
  return x;
}
int bitcnt(int x) {
  x = ((x & 0xAAAAAAAA) >> 1) + (x & 0x55555555);
  x = ((x & 0xCCCCCCCC) >> 2) + (x & 0x33333333);
  x = ((x & 0xF0F0F0F0) >> 4) + (x & 0x0F0F0F0F);
  x = ((x & 0xFF00FF00) >> 8) + (x & 0x00FF00FF);
  x = ((x & 0xFFFF0000) >> 16) + (x & 0x0000FFFF);
  return x;
}
ll sqrt(ll x) {
  ll left = 0, right = x;
  rep(i, 100) {
    ll mid = (left + right) >> 1;
    if (mid * mid <= x)
      left = mid;
    else
      right = mid;
  }
  return left;
}
ll gcd(ll a, ll b) { return a % b == 0 ? b : gcd(b, a % b); }
#define debug(x) printf("Case #%d: ", x)
#define DEBUG 0
const ll inf = 1e18;
const ll mod = 1e9 + 7;
const ld eps = 1e-15;
const int dx[] = {1, 0, -1, 0, 0};
const int dy[] = {0, 1, 0, -1, 0};

// 取り敢えず平均がK→各要素Kずつ引いて、合計が非負ならOKと見ます。
// そうすると、累積和を取っておくと区間和が各区間についてO(1)、全体でO(N)で計算できます。ただ、このままでは間に合いません。
// ここで、累積和をsum[i]という配列に入れたとしたとき、「区間和の右側はsum[i]以下であれば全部カウントしてよい」という特徴があります。
// となると、左から見ていき、今まで出てきたsumを記録しておいて、新しいsum[i]に対してsum[i]が何番目に相当するかを調べることが出来れば解けそうです。
// 具体的には、記録した見ている箇所より左の集合の中で、「sum[i]より小さなもの」の個数がそこを左端とする個数です。（右側を探しているので、引くことに注意）
// 私はこれをBitIndexedTree(多分分科会であとでやります）を用いて実装しました。vectorは挿入に実は最悪O(N)かかるので、結構大変なので。
// 他の実装方法もあると思うので、頑張ってみてください。

// Bit Indexed Tree
// 0は使わないので注意。
int bit[400010];
int N;

void init(int n) {
  N = 1;
  while (N <= n)
    N <<= 1;
}

void update(int k, int a) {
  while (k < N) {
    bit[k] += a;
    k += k & -k;
  }
}

int getsum(int k) {
  int ret = 0;
  while (k > 0) {
    ret += bit[k];
    k -= k & -k;
  }
  return ret;
}

// 当然ですが、答えはint型に収まらないことがあるので注意

int main() {
  int n, k;
  cin >> n >> k;
  vector<ll> a(n);
  rep(i, n) {
    cin >> a[i];
    a[i] -= k;
  }
  vector<ll> sum(n + 1);
  SREP(i, 1, n) { sum[i] = sum[i - 1] + a[i - 1]; }
  // bit用にソートした累積和配列を用意
  // aの値そのままでbitをやると死ぬので、座標圧縮をします。
  vector<ll> sorted = sum;
  sort(all(sorted));

  init(n + 1);
  ll ret = 0;
  REP(i, n) {
    // sum[i]の位置を調べます。
    int addidx = lower_bound(all(sorted), sum[i]) - sorted.begin();
    // そこより左の個数をカウント（自身は含めません）
    ret += getsum(addidx + 1);
    // sum[i]を追加します。
    update(addidx + 1, 1);
  }
  cout << ret << endl;
  return 0;
}