結果
問題 | No.1115 二つの数列 / Two Sequences |
ユーザー |
![]() |
提出日時 | 2020-07-20 15:08:09 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 191 ms / 2,000 ms |
コード長 | 2,207 bytes |
コンパイル時間 | 2,268 ms |
コンパイル使用メモリ | 180,960 KB |
実行使用メモリ | 13,952 KB |
最終ジャッジ日時 | 2024-12-23 18:42:31 |
合計ジャッジ時間 | 7,096 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long ll; #define rep(i, n) for(ll i = 0, i##_len = (n); i < i##_len; i++) #define reps(i, s, n) for(ll i = (s), i##_len = (n); i < i##_len; i++) #define rrep(i, n) for(ll i = (n) - 1; i >= 0; i--) #define rreps(i, e, n) for(ll i = (n) - 1; i >= (e); i--) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define sz(x) ((ll)(x).size()) #define len(x) ((ll)(x).length()) #define endl "\n" template<class T> struct SegmentTree { private: int n; vector<T> node; T def; function<T(T,T)> lop; function<T(T,T)> out; public: SegmentTree() {} SegmentTree(const vector<T> &v, T _def, function<T(T,T)> _lop, function<T(T,T)> _out) { int vl = (int)v.size(); n = 1; while(n < vl) n *= 2; def = _def; lop = _lop; out = _out; node = vector<T>(2 * n - 1, def); for(int i = 0; i < vl; i++) node[i + n - 1] = v[i]; for(int i = n - 2; i >= 0; i--) node[i] = out(node[i * 2 + 1], node[i * 2 + 2]); } void update(int idx, T val) { idx += (n - 1); node[idx] = lop(node[idx], val); while(idx > 0) { idx = (idx - 1) / 2; node[idx] = out(node[idx * 2 + 1], node[idx * 2 + 2]); } } T query(int a, int b, int k = 0, int l = 0, int r = -1) { if (r < 0) r = n; if ((r <= a) || (b <= l)) return def; if ((a <= l) && (r <= b)) return node[k]; return out(query(a, b, 2 * k + 1, l, (l + r) / 2), query(a, b, 2 * k + 2, (l + r) / 2, r)); } }; int main() { cin.tie(0); ios::sync_with_stdio(false); // ifstream in("input.txt"); // cin.rdbuf(in.rdbuf()); ll n; cin >> n; vector<ll> a(n), b(n); rep(i, n) cin >> a[i]; rep(i, n) cin >> b[i]; map<ll, ll> idx; rep(i, n) idx[a[i]] = i; auto lop = [](ll a, ll b){ return b; }; auto out = [](ll a, ll b){ return a + b; }; SegmentTree<ll> st(vector<ll>(n, 0), 0, lop, out); ll ans = 0; rep(i, n) { ll cnt = st.query(idx[b[i]], n); st.update(idx[b[i]], 1); ans += cnt; } cout << ans << endl; return 0; }