京都大学プログラミングコンテスト 2015 G: ケンドー

問題

G: ケンドー - 京都大学プログラミングコンテスト2015 | AtCoder

解法

Bは昇順でソートされているので後ろから見ていき、B[i]より大きいAの中で最も右側にあるものをB[i]と戦わせていけば良い。
i を回して、 Priority Queue に毎回 B[i] 以上のA を入れていけば、キューを B[i] に負けないAの集合として扱うことが出来る。
入れ替え回数を計算するとき、自分より前に居たAが何個抜けているかをBITで管理すると速くなる。

コード

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

template <typename T>
class FenwickTree {
 private:
  int N;
  vector<T> dat;

 public:
  FenwickTree(int N) : N(N) { dat.assign(N, 0); }

  void add(int k, T val) {
    for (int x = k; x < N; x |= x + 1) {
      dat[x] += val;
    }
  }

  // [0, k)
  T sum(int k) {
    T ret = 0;
    for (int x = k - 1; x >= 0; x = (x & (x + 1)) - 1) {
      ret += dat[x];
    }
    return ret;
  }

  // [l, r)
  T sum(int l, int r) { return sum(r) - sum(l); }

  T get(int k) {
    assert(0 <= k && k < N);
    return sum(k + 1) - sum(k);
  }
};

int main() {
  int N;
  cin >> N;

  vector<int> A(N);
  for (int i = 0; i < N; ++i) cin >> A[i];
  vector<int> B(N);
  for (int i = 0; i < N; ++i) cin >> B[i];

  vector<pair<int, int>> sortByPlace(N);
  for (int i = 0; i < N; ++i) {
    sortByPlace[i] = {A[i], i};
  }

  sort(sortByPlace.begin(), sortByPlace.end());

  FenwickTree<int> bit(N);
  priority_queue<pair<int, int>> que;
  ll ans = 0;
  for (int i = N - 1, j = N - 1; i >= 0; --i) {
    while (j >= 0 && sortByPlace[j].first >= B[i]) {
      que.push({sortByPlace[j].second, sortByPlace[j].first});
      j--;
    }
    if (que.empty()) {
      cout << "-1" << endl;
      return 0;
    }

    int place = que.top().first;
    que.pop();
    ans += i - place + bit.sum(place);
    bit.add(place, 1);
  }
  cout << ans << endl;
}