結果
問題 |
No.1209 XOR Into You
|
ユーザー |
|
提出日時 | 2020-10-16 14:53:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 253 ms / 2,000 ms |
コード長 | 1,987 bytes |
コンパイル時間 | 2,193 ms |
コンパイル使用メモリ | 213,684 KB |
最終ジャッジ日時 | 2025-01-15 07:48:14 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 37 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:65:21: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 65 | int n; scanf("%d",&n); | ~~~~~^~~~~~~~~ main.cpp:67:23: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 67 | rep(i,n) scanf("%d",&a[i]); | ~~~~~^~~~~~~~~~~~ main.cpp:68:23: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 68 | rep(i,n) scanf("%d",&b[i]); | ~~~~~^~~~~~~~~~~~
ソースコード
#include <bits/stdc++.h> #define rep(i,n) for(int i=0;i<(n);i++) using namespace std; template<class G> class Fenwick_tree{ vector<G> a; public: Fenwick_tree(){} Fenwick_tree(int n){ build(n); } Fenwick_tree(const vector<G>& a){ build(a); } void build(int n){ a.assign(n,G{}); } void build(const vector<G>& a){ this->a=a; for(int i=1;i<a.size();i++) if(i+(i&-i)-1<a.size()) (this->a)[i+(i&-i)-1]+=(this->a)[i-1]; } void add(int i,const G& val){ for(i++;i<=a.size();i+=i&-i) a[i-1]+=val; } G sum(int l,int r)const{ if(l==0){ G res{}; for(;r>0;r-=r&-r) res+=a[r-1]; return res; } return sum(0,r)-sum(0,l); } int lower_bound(G val)const{ if(val<=G{}) return 0; int x=0,k; for(k=1;k<=a.size();k<<=1); for(k>>=1;k>0;k>>=1) if(x+k<=a.size() && a[x+k-1]<val) val-=a[x+k-1], x+=k; return x; } int upper_bound(G val)const{ if(val<G{}) return 0; int x=0,k; for(k=1;k<=a.size();k<<=1); for(k>>=1;k>0;k>>=1) if(x+k<=a.size() && a[x+k-1]<=val) val-=a[x+k-1], x+=k; return x; } }; template<class T> long long inversion_number(const vector<T>& a){ auto X=a; sort(X.begin(),X.end()); X.erase(unique(X.begin(),X.end()),X.end()); long long res=0; Fenwick_tree<int> F(X.size()); rep(i,a.size()){ int x=lower_bound(X.begin(),X.end(),a[i])-X.begin(); res+=F.sum(x+1,X.size()); F.add(x,1); } return res; } int main(){ int n; scanf("%d",&n); vector<int> a(n),b(n); rep(i,n) scanf("%d",&a[i]); rep(i,n) scanf("%d",&b[i]); vector<int> a2(n),b2(n); a2[0]=a[0]; b2[0]=b[0]; rep(i,n-1){ a2[i+1]=a[i]^a[i+1]; b2[i+1]=b[i]^b[i+1]; } { auto a3=a2; auto b3=b2; sort(a3.begin(),a3.end()); sort(b3.begin(),b3.end()); if(a3!=b3){ puts("-1"); return 0; } } map<int,int> f; map<pair<int,int>,int> g; rep(i,n){ int cnt=f[a2[i]]; g[{a2[i],cnt}]=i; ++f[a2[i]]; } f.clear(); rep(i,n){ int cnt=f[b2[i]]; ++f[b2[i]]; b2[i]=g[{b2[i],cnt}]; } printf("%lld\n",inversion_number(b2)); return 0; }