Problem p03703 (C++)

Buggy submission (status: Runtime Error):
// This amazing code is by Eric Sunli Chen.
#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <utility>
#include <vector>
using namespace std;
template <typename T> void get_int(T &x) {
  char t = getchar();
  bool neg = false;
  x = 0;
  for (; (t > '9' || t < '0') && t != '-'; t = getchar())
    ;
  if (t == '-')
    neg = true, t = getchar();
  for (; t <= '9' && t >= '0'; t = getchar())
    x = x * 10 + t - '0';
  if (neg)
    x = -x;
}
template <typename T> void print_int(T x) {
  if (x < 0)
    putchar('-'), x = -x;
  short a[20] = {}, sz = 0;
  while (x > 0)
    a[sz++] = x % 10, x /= 10;
  if (sz == 0)
    putchar('0');
  for (int i = sz - 1; i >= 0; i--)
    putchar('0' + a[i]);
}
#define ff first
#define ss second
#define pb push_back
#define mp make_pair
#define get1(a) get_int(a)
#define get2(a, b) get1(a), get1(b)
#define get3(a, b, c) get1(a), get2(b, c)
#define printendl(a) print_int(a), puts("")
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<int, int> pii;
const int inf = 0x3f3f3f3f;
const LL Linf = 1ll << 61;
const double pi = acos(-1.0);

vector<LL> v;
int n, k, a[100111], bit[200111];
void add(int x) {
  for (; x < 200111; x += x & -x)
    bit[x]++;
}
int query(int x) {
  int ret = 0;
  for (; x > 0; x -= x & -x)
    ret += bit[x];
  return ret;
}
int getid(LL x) { return lower_bound(v.begin(), v.end(), x) - v.begin() + 1; }
int main() {
  get2(n, k);
  for (int i = 1; i <= n; i++) {
    get1(a[i]);
    a[i] -= k;
  }
  LL sum = 0;
  v.pb(sum);
  for (int i = 1; i <= n; i++) {
    sum += a[i];
    v.pb(sum);
  }
  sort(v.begin(), v.end());
  sum = 0;
  add(getid(0));
  LL ans = 0;
  for (int i = 1; i <= n; i++) {
    sum += a[i];
    ans += query(getid(sum));
    add(getid(sum));
  }
  printendl(ans);
  return 0;
}

Fixed submission:
// This amazing code is by Eric Sunli Chen.
#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <utility>
#include <vector>
using namespace std;
template <typename T> void get_int(T &x) {
  char t = getchar();
  bool neg = false;
  x = 0;
  for (; (t > '9' || t < '0') && t != '-'; t = getchar())
    ;
  if (t == '-')
    neg = true, t = getchar();
  for (; t <= '9' && t >= '0'; t = getchar())
    x = x * 10 + t - '0';
  if (neg)
    x = -x;
}
template <typename T> void print_int(T x) {
  if (x < 0)
    putchar('-'), x = -x;
  short a[20] = {}, sz = 0;
  while (x > 0)
    a[sz++] = x % 10, x /= 10;
  if (sz == 0)
    putchar('0');
  for (int i = sz - 1; i >= 0; i--)
    putchar('0' + a[i]);
}
#define ff first
#define ss second
#define pb push_back
#define mp make_pair
#define get1(a) get_int(a)
#define get2(a, b) get1(a), get1(b)
#define get3(a, b, c) get1(a), get2(b, c)
#define printendl(a) print_int(a), puts("")
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<int, int> pii;
const int inf = 0x3f3f3f3f;
const LL Linf = 1ll << 61;
const double pi = acos(-1.0);

vector<LL> v;
int n, k, a[200111], bit[200111];
void add(int x) {
  for (; x < 200111; x += x & -x)
    bit[x]++;
}
int query(int x) {
  int ret = 0;
  for (; x > 0; x -= x & -x)
    ret += bit[x];
  return ret;
}
int getid(LL x) { return lower_bound(v.begin(), v.end(), x) - v.begin() + 1; }
int main() {
  get2(n, k);
  for (int i = 1; i <= n; i++) {
    get1(a[i]);
    a[i] -= k;
  }
  LL sum = 0;
  v.pb(sum);
  for (int i = 1; i <= n; i++) {
    sum += a[i];
    v.pb(sum);
  }
  sort(v.begin(), v.end());
  sum = 0;
  add(getid(0));
  LL ans = 0;
  for (int i = 1; i <= n; i++) {
    sum += a[i];
    ans += query(getid(sum));
    add(getid(sum));
  }
  printendl(ans);
  return 0;
}