結果
問題 | No.2732 Similar Permutations |
ユーザー |
|
提出日時 | 2024-04-19 23:22:22 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,077 ms / 2,000 ms |
コード長 | 8,911 bytes |
コンパイル時間 | 3,524 ms |
コンパイル使用メモリ | 247,424 KB |
実行使用メモリ | 261,300 KB |
最終ジャッジ日時 | 2024-10-11 17:58:00 |
合計ジャッジ時間 | 22,899 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 101 |
ソースコード
#include<bits/stdc++.h> using namespace std; using ll=long long; int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int n; cin>>n; if(n==1){ cout<<-1<<endl; return 0; } vector<int> a(n); map<int,int> cnt; for(int i=0;i<n;i++) cin>>a[i],cnt[a[i]]++; if(cnt.size()<n){ vector<int> p(n); for(int i=0;i<n;i++) p[i]=i+1; for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; int x=-1; for(auto& [key,val]:cnt){ if(val>1){ x=key; } } vector<int> idx; for(int i=0;i<n;i++) if(a[i]==x){ idx.push_back(i); } swap(p[idx[0]],p[idx[1]]); for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; return 0; } vector<int> p(n); if(n<=10){ for(int i=0;i<n;i++) p[i]=i+1; map<int,vector<vector<int>>> mp; do{ int Xor=0; for(int i=0;i<n;i++) Xor^=a[i]+p[i]; mp[Xor].push_back(p); }while(next_permutation(p.begin(),p.end())); for(auto& [Xor,vecs]:mp){ if(vecs.size()<2) continue; for(int i=0;i<n;i++) cout<<vecs[0][i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<vecs[1][i]<<' '; cout<<endl; return 0; } cout<<-1<<endl; return 0; } vector<pair<int,int>> vec(n); for(int i=0;i<n;i++){ vec[i]={a[i],i}; } sort(vec.rbegin(),vec.rend()); for(int i=0;i<n;i++){ p[vec[i].second]=i+1; } if(n<=5000){ for(int i=0;i<n;i++){ for(int j=i+1;j<n;j++){ if(((a[i]+p[i])^(a[j]+p[j]))==((a[i]+p[j])^(a[j]+p[i]))){ vector<int> q=p; swap(q[i],q[j]); for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } } } } vector<int> ip(n); for(int i=0;i<n;i++) ip[p[i]-1]=i; vector<int> q=p; swap(q[ip[0]],q[ip[1]]); int x=0,y=0; for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } if(n==2){ cout<<-1<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[2]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[1]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[2]],q[ip[1]]); swap(q[ip[2]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[1]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } random_device seed_gen; mt19937 engine(seed_gen()); shuffle(p.begin(), p.end(), engine); for(int i=0;i<n;i++) ip[p[i]-1]=i; q=p; swap(q[ip[0]],q[ip[1]]); x=0,y=0; for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } if(n==2){ cout<<-1<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[2]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[1]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[2]],q[ip[1]]); swap(q[ip[2]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[1]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } shuffle(p.begin(), p.end(), engine); for(int i=0;i<n;i++) ip[p[i]-1]=i; q=p; swap(q[ip[0]],q[ip[1]]); x=0,y=0; for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } if(n==2){ cout<<-1<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[2]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[1]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[2]],q[ip[1]]); swap(q[ip[2]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[1]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } shuffle(p.begin(), p.end(), engine); for(int i=0;i<n;i++) ip[p[i]-1]=i; q=p; swap(q[ip[0]],q[ip[1]]); x=0,y=0; for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } if(n==2){ cout<<-1<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[2]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[1]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[2]],q[ip[1]]); swap(q[ip[2]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[1]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } shuffle(p.begin(), p.end(), engine); for(int i=0;i<n;i++) ip[p[i]-1]=i; q=p; swap(q[ip[0]],q[ip[1]]); x=0,y=0; for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } if(n==2){ cout<<-1<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[2]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[1]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[2]],q[ip[1]]); swap(q[ip[2]],q[ip[0]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } x=0,y=0; swap(q[ip[0]],q[ip[1]]); for(int i=0;i<n;i++) x^=a[i]+p[i]; for(int i=0;i<n;i++) y^=a[i]+q[i]; if(x==y){ for(int i=0;i<n;i++) cout<<p[i]<<' '; cout<<endl; for(int i=0;i<n;i++) cout<<q[i]<<' '; cout<<endl; return 0; } cout<<-1<<endl; }