結果
問題 | No.1115 二つの数列 / Two Sequences |
ユーザー |
![]() |
提出日時 | 2020-08-06 03:22:37 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 72 ms / 2,000 ms |
コード長 | 2,153 bytes |
コンパイル時間 | 894 ms |
コンパイル使用メモリ | 96,956 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-18 23:56:45 |
合計ジャッジ時間 | 3,337 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 35 |
ソースコード
#include <algorithm> #include <cmath> #include <cstdint> #include <cstdio> #include <iomanip> #include <iostream> #include <map> #include <queue> #include <set> #include <stack> #include <string> #include <vector> using namespace std; #define rep(i, n) for(int(i) = 0; (i) < (n); (i)++) #define FOR(i, m, n) for(int(i) = (m); (i) < (n); (i)++) #define All(v) (v).begin(), (v).end() #define pb push_back #define MP(a, b) make_pair((a), (b)) using ll = long long; using pii = pair<int, int>; using pll = pair<ll, ll>; const int INF = 1 << 30; const ll LINF = 1LL << 60; const int MOD = 1e9 + 7; // 参考:https://qiita.com/drken/items/1b7e6e459c24a83bb7fd // 1-indexed BIT // add:log(N),sum:log(N),get_k-th:log(N) // https://codeforces.com/contest/1354/problem/D struct BIT { vector<ll> dat; BIT(int n) { dat.resize(n + 1); for(int i = 0; i < (int)dat.size(); i++) dat[i] = 0; } // a is 1-indexed void add(int a, ll x) { for(int i = a; i < (int)dat.size(); i += i & -i) dat[i] = dat[i] + x; } // [1,a], a is 1-indexed ll sum(int a) { ll res = 0; for(int i = a; i > 0; i -= i & -i) { res = res + dat[i]; } return res; } // [a,b), a and b are 1-indexed ll sum(int a, int b) { return sum(b - 1) - sum(a - 1); } // k-th number (k is 1-indexed) // lower_bound int get(ll k) { if(k <= 0) return 0; int res = 0; int N = 1; while(N < (int)dat.size()) N *= 2; for(int i = N / 2; i > 0; i /= 2) { if(res + i < (int)dat.size() && dat[res + i] < k) { k = k - dat[res + i]; res = res + i; } } return res + 1; } }; int main() { int N; cin >> N; vector<ll> A(N), B(N); vector<ll> idx(N + 1); rep(i, N) { cin >> A[i]; idx[A[i]] = i + 1; } rep(i, N) cin >> B[i]; BIT bit(N + 1); ll res = 0; rep(i, N) { res += i - bit.sum(idx[B[i]]); bit.add(idx[B[i]], 1); } cout << res << endl; return 0; }