結果
問題 |
No.1115 二つの数列 / Two Sequences
|
ユーザー |
![]() |
提出日時 | 2020-07-19 02:07:27 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 66 ms / 2,000 ms |
コード長 | 2,103 bytes |
コンパイル時間 | 2,049 ms |
コンパイル使用メモリ | 198,524 KB |
最終ジャッジ日時 | 2025-01-12 00:41:03 |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h> using namespace std; template<typename T> class FenwickTree { private: int n, mx; vector<T> bit; public: FenwickTree(int sz) : n(sz + 1), mx(1), bit(n, 0) { while (mx * 2 <= n) mx *= 2; } FenwickTree(const vector<T> &v) : n((int) v.size() + 1), mx(1), bit(n, 0) { for (int i = 0; i < n - 1; i++) add(i, v.at(i)); while (mx * 2 <= n) mx *= 2; } void add(int i, T x) { i++; while (i < n) bit.at(i) += x, i += i & -i; } void add(int l, int r, T x) { add(l, x); add(r + 1 , -x); } T sum(int i) { i++; T ret = 0; while (i > 0) ret += bit.at(i), i -= i & -i; return ret; } T sum(int l, int r) { return sum(r) - sum(l - 1); } int lower_bound(T w) { if (w <= 0) return 0; int x = 0; for (int k = mx; k > 0; k /= 2) { if (x + k <= n - 1 && bit.at(x + k) < w) { w -= bit.at(x + k); x += k; } } return x; } int upper_bound(T w) { if (w < 0) return 0; int x = 0; for (int k = mx; k > 0; k /= 2) { if (x + k <= n - 1 && bit.at(x + k) <= w) { w -= bit.at(x + k); x += k; } } return x; } }; long inv_count(const vector<int> &u) { int n = (int) u.size(); FenwickTree<int> FT(n); long ans = 0; for (int i = 0; i < n; i++) { ans += i - FT.sum(u.at(i)); FT.add(u.at(i), 1); } return ans; } long inv_count(const vector<int> &u, const vector<int> &v) { int n = (int) u.size(); vector<vector<int>> p(n); FenwickTree<int> FT(n); for (int i = n - 1; i >= 0; i--) { p.at(u.at(i)).push_back(i); } long ans = 0; for (int i = 0; i < n; ++i) { int pos = p.at(v.at(i)).back(); p.at(v.at(i)).pop_back(); ans += pos - FT.sum(pos); FT.add(pos, 1); } return ans; } int main() { cin.tie(nullptr); ios::sync_with_stdio(false); int N; cin >> N; vector<int> A(N); for (int i = 0; i < N; i++) { cin >> A.at(i); A.at(i)--; } vector<int> B(N); for (int i = 0; i < N; i++) { cin >> B.at(i); B.at(i)--; } cout << inv_count(A, B) << "\n"; }