結果
問題 | No.1115 二つの数列 / Two Sequences |
ユーザー |
|
提出日時 | 2020-09-26 14:54:11 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 28 ms / 2,000 ms |
コード長 | 4,312 bytes |
コンパイル時間 | 1,648 ms |
コンパイル使用メモリ | 168,688 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-06-29 02:16:17 |
合計ジャッジ時間 | 4,258 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h>#define all(v) v.begin(), v.end()#define rall(v) v.rbegin(), v.rend()#define rep(i,n) for (int i=0;i<(int)(n);i++)#define codefor int test;scanf("%d",&test);while(test--)#define INT(...) int __VA_ARGS__;in(__VA_ARGS__)#define LL(...) ll __VA_ARGS__;in(__VA_ARGS__)#define yes(ans) if(ans)printf("yes\n");else printf("no\n")#define Yes(ans) if(ans)printf("Yes\n");else printf("No\n")#define YES(ans) if(ans)printf("YES\n");else printf("NO\n")#define vector1d(type,name,...) vector<type>name(__VA_ARGS__)#define vector2d(type,name,h,...) vector<vector<type>>name(h,vector<type>(__VA_ARGS__))#define umap unordered_map#define uset unordered_setusing namespace std;using ll = long long;const int MOD=1000000007;const int MOD2=998244353;const int INF=1<<30;const ll INF2=(ll)1<<60;//入力系void scan(int& a){scanf("%d",&a);}void scan(long long& a){scanf("%lld",&a);}template<class T,class L>void scan(pair<T, L>& p){scan(p.first);scan(p.second);}template<class T> void scan(T& a){cin>>a;}template<class T> void scan(vector<T>& vec){for(auto&& it:vec)scan(it);}void in(){}template <class Head, class... Tail> void in(Head& head, Tail&... tail){scan(head);in(tail...);}//出力系void print(const int& a){printf("%d",a);}void print(const long long& a){printf("%lld",a);}void print(const double& a){printf("%.15lf",a);}template<class T,class L>void print(const pair<T, L>& p){print(p.first);putchar(' ');print(p.second);}template<class T> void print(const T& a){cout<<a;}template<class T> void print(const vector<T>& vec){if(vec.empty())return;print(vec[0]);for(auto it=vec.begin();++it!= vec.end();){putchar(' ');print(*it);}}void out(){putchar('\n');}template<class T> void out(const T& t){print(t);putchar('\n');}template <class Head, class... Tail> void out(const Head& head,const Tail&... tail){print(head);putchar(' ');out(tail...);}//デバッグ系template<class T> void dprint(const T& a){cerr<<a;}template<class T> void dprint(const vector<T>& vec){if(vec.empty())return;cerr<<vec[0];for(auto it=vec.begin();++it!= vec.end();){cerr<<" "<<*it;}}void debug(){cerr<<endl;}template<class T> void debug(const T& t){dprint(t);cerr<<endl;}template <class Head, class... Tail> void debug(const Head& head, const Tail&... tail){dprint(head);cerr<<" ";debug(tail...);}ll intpow(ll a, ll b){ ll ans = 1; while(b){ if(b & 1) ans *= a; a *= a; b /= 2; } return ans; }ll modpow(ll a, ll b, ll p){ ll ans = 1; while(b){ if(b & 1) (ans *= a) %= p; (a *= a) %= p; b /= 2; } return ans; }ll updivide(ll a,ll b){if(a%b==0) return a/b;else return (a/b)+1;}template <class T> struct fenwick_tree {using U = T;public:fenwick_tree() : _n(0) {}fenwick_tree(int n) : _n(n), data(n) {}void add(int p, T x) {assert(0 <= p && p < _n);p++;while (p <= _n) {data[p - 1] += U(x);p += p & -p;}}T sum(int l, int r) {assert(0 <= l && l <= r && r <= _n);return sum(r) - sum(l);}private:int _n;std::vector<U> data;U sum(int r) {U s = 0;while (r > 0) {s += data[r - 1];r -= r & -r;}return s;}};int main(){INT(n);vector<int> a(n),b(n);int x;rep(i,n){in(x);x--;a[i]=x;}rep(i,n){in(x);x--;b[x]=i;}//iがaの場所//b[i]がbの場所rep(i,n){//a[i]はbの場所//a={1,5,4,2,3}//b={2,1,4,3,5}//ならばa={1,4,2,0,3}となるa[i]=b[a[i]];}fenwick_tree<int> fw(n);ll ans=0;for(int i=0;i<n;i++){//a={1,4,2,0,3}についての操作を示す//1回目のループ//ans+=0-sum[0,1)=0-0 →ans=0//fw={0,1,0,0,0}//2回目のループ//ans+=1-sum[0,4)=1-1 → ans=0//fw={0,1,0,0,1}//3回目のループ//ans+=2-sum[0,2)=2-1 → ans=1//fw={0,1,1,0,1}//4回目のループ//ans+=3-sum[0,0)=3-0 → ans=4//fw={0,1,1,1,1}//5回目のループ//ans+=4-sum[0,3)=3-2 → ans=1//fw={1,1,1,1,1}ans+=i-fw.sum(0,a[i]);fw.add(a[i], 1);}out(ans);}