結果
問題 |
No.1115 二つの数列 / Two Sequences
|
ユーザー |
![]() |
提出日時 | 2020-09-25 16:03:54 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 54 ms / 2,000 ms |
コード長 | 2,294 bytes |
コンパイル時間 | 1,345 ms |
コンパイル使用メモリ | 116,124 KB |
実行使用メモリ | 7,296 KB |
最終ジャッジ日時 | 2024-06-28 05:49:34 |
合計ジャッジ時間 | 4,553 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 35 |
ソースコード
#pragma GCC optimize("O3") #include <iostream> #include <iomanip> #include <cstdio> #include <string> #include <cstring> #include <deque> #include <list> #include <queue> #include <stack> #include <vector> #include <utility> #include <algorithm> #include <map> #include <set> #include <complex> #include <cmath> #include <limits> #include <cfloat> #include <climits> #include <ctime> #include <cassert> #include <numeric> #include <fstream> #include <functional> #include <bitset> using namespace std; using ll = long long; using P = pair<int, int>; using T = tuple<int, int, int>; template <class T> inline T chmax(T &a, const T b) {return a = (a < b) ? b : a;} template <class T> inline T chmin(T &a, const T b) {return a = (a > b) ? b : a;} constexpr int MOD = 1e9 + 7; constexpr int inf = 1e9; constexpr long long INF = 1e18; #define all(a) (a).begin(), (a).end() int dx[] = {1, 0, -1, 0}; int dy[] = {0, 1, 0, -1}; class SegmentTree{ public: int n; vector<ll> node; SegmentTree(vector<ll> v) { int sz = v.size(); n = 1; while(n < sz) n *= 2; node.resize(2*n-1, 0); for(int i=0; i<sz; i++) node[i+n-1] = v[i]; for(int i=n-2; i>=0; i--) node[i] = node[i*2+1] + node[i*2+2]; } void add(int x, ll val){ x += (n - 1); node[x] += val; while(0 < x){ x = (x - 1) / 2; node[x] = node[2*x+1] + node[2*x+2]; } } ll getsum(int a, int b, int k=0, int l=0, int r=-1){ if(r < 0) r = n; if(b <= l || r <= a) return 0; if(a <= l && r <= b) return node[k]; ll vl = getsum(a, b, 2*k+1, l, (l+r)/2); ll vr = getsum(a, b, 2*k+2, (l+r)/2, r); return vl + vr; } }; int main(){ cin.tie(0); ios::sync_with_stdio(false); int n; cin>>n; vector<int> a(n), b(n); for(int i=0; i<n; i++){ cin>>a[i]; a[i]--; } for(int i=0; i<n; i++){ cin>>b[i]; b[i]--; } vector<int> atoidx(n); for(int i=0; i<n; i++){ atoidx[a[i]] = i; } SegmentTree seg(vector<ll>(n, 0)); ll ans = 0; for(int i=0; i<n; i++){ int idx = atoidx[b[i]]; ans += seg.getsum(idx, n); seg.add(idx, 1); } cout << ans << endl; return 0; }