Problem p03357 (C++)

Buggy submission (status: Runtime Error):
#include <iostream>
#include <string>
#include <utility>
#include <vector>
#define llint long long
#define inf 1e9

using namespace std;
typedef pair<llint, llint> P;

struct BIT {
  int size;
  vector<llint> bit;
  BIT() { size = 0; }
  BIT(int s) {
    size = s;
    bit.resize(size + 1);
    init();
  }
  void init() {
    for (int i = 1; i <= size; i++)
      bit[i] = 0;
  }
  llint query(int i) {
    llint ret = 0;
    while (i > 0) {
      ret += bit[i];
      i -= i & (-i);
    }
    return ret;
  }
  void add(int i, llint x) {
    while (i <= size) {
      bit[i] += x;
      i += i & (-i);
    }
  }
};

llint n;
char c[2005];
int a[2005];
int bpos[2005], wpos[2005];
llint dp[2005][2005];
BIT bbit(4005), wbit(4005);

int main(void) {
  ios::sync_with_stdio(0);
  cin.tie(0);

  cin >> n;
  for (int i = 1; i <= 2 * n; i++)
    cin >> c[i] >> a[i];

  for (int i = 1; i <= 2 * n; i++) {
    if (c[i] == 'B')
      bpos[a[i]] = i;
    else
      wpos[a[i]] = i;
  }

  for (int i = 0; i <= n; i++) {
    for (int j = 0; j <= n; j++) {
      dp[i][j] = inf;
    }
  }
  dp[0][0] = 0;

  for (int i = 0; i <= n; i++) {
    wbit.init();
    for (int j = 0; j <= n; j++) {
      if (i == n && j == n)
        continue;
      if (i + 1 <= n)
        dp[i + 1][j] = min(dp[i + 1][j], dp[i][j] + (bpos[i + 1] - 1) -
                                             bbit.query(bpos[i + 1]) -
                                             wbit.query(bpos[i + 1]));
      if (j + 1 <= n)
        dp[i][j + 1] = min(dp[i][j + 1], dp[i][j] + (wpos[j + 1] - 1) -
                                             bbit.query(wpos[j + 1]) -
                                             wbit.query(wpos[j + 1]));
      if (j + 1 <= n)
        wbit.add(wpos[j + 1], 1);
    }
    if (i + 1 <= n)
      bbit.add(bpos[i + 1], 1);
  }
  cout << dp[n][n] << endl;

  return 0;
}

Fixed submission:
#include <iostream>
#include <string>
#include <utility>
#include <vector>
#define llint long long
#define inf 1e9

using namespace std;
typedef pair<llint, llint> P;

struct BIT {
  int size;
  vector<llint> bit;
  BIT() { size = 0; }
  BIT(int s) {
    size = s;
    bit.resize(size + 1);
    init();
  }
  void init() {
    for (int i = 1; i <= size; i++)
      bit[i] = 0;
  }
  llint query(int i) {
    llint ret = 0;
    while (i > 0) {
      ret += bit[i];
      i -= i & (-i);
    }
    return ret;
  }
  void add(int i, llint x) {
    while (i <= size) {
      bit[i] += x;
      i += i & (-i);
    }
  }
};

llint n;
char c[4005];
int a[4005];
int bpos[2005], wpos[2005];
llint dp[2005][2005];
BIT bbit(4005), wbit(4005);

int main(void) {
  ios::sync_with_stdio(0);
  cin.tie(0);

  cin >> n;
  for (int i = 1; i <= 2 * n; i++)
    cin >> c[i] >> a[i];

  for (int i = 1; i <= 2 * n; i++) {
    if (c[i] == 'B')
      bpos[a[i]] = i;
    else
      wpos[a[i]] = i;
  }

  for (int i = 0; i <= n; i++) {
    for (int j = 0; j <= n; j++) {
      dp[i][j] = inf;
    }
  }
  dp[0][0] = 0;

  for (int i = 0; i <= n; i++) {
    wbit.init();
    for (int j = 0; j <= n; j++) {
      if (i == n && j == n)
        continue;
      if (i + 1 <= n)
        dp[i + 1][j] = min(dp[i + 1][j], dp[i][j] + (bpos[i + 1] - 1) -
                                             bbit.query(bpos[i + 1]) -
                                             wbit.query(bpos[i + 1]));
      if (j + 1 <= n)
        dp[i][j + 1] = min(dp[i][j + 1], dp[i][j] + (wpos[j + 1] - 1) -
                                             bbit.query(wpos[j + 1]) -
                                             wbit.query(wpos[j + 1]));
      if (j + 1 <= n)
        wbit.add(wpos[j + 1], 1);
    }
    if (i + 1 <= n)
      bbit.add(bpos[i + 1], 1);
  }
  cout << dp[n][n] << endl;

  return 0;
}