京都大学プログラミングコンテスト 2015 G: ケンドー
解法
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; }