結果
問題 | No.2732 Similar Permutations |
ユーザー |
![]() |
提出日時 | 2024-04-19 21:50:40 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 166 ms / 2,000 ms |
コード長 | 2,115 bytes |
コンパイル時間 | 4,484 ms |
コンパイル使用メモリ | 265,136 KB |
最終ジャッジ日時 | 2025-02-21 04:07:54 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 101 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf32 1000000001 #define Inf64 1000000000000000001 void Out(vector<int> p){ rep(i,p.size()){ if(i!=0)cout<<' '; cout<<p[i]; } cout<<endl; } int main(){ int n; cin>>n; vector<int> a(n); rep(i,n)cin>>a[i]; if(n<=8){ vector<int> p(n); rep(i,n)p[i] = i+1; map<int,vector<int>> mp; do{ int s = 0; rep(i,n)s ^= a[i] + p[i]; if(mp.count(s)){ Out(p); Out(mp[s]); return 0; } else mp[s] = p; } while(next_permutation(p.begin(),p.end())); cout<<-1<<endl; } else{ rep(i,n)a[i] ++; vector<vector<int>> is(4); rep(i,n)is[a[i]%4].push_back(i); vector<int> p(n,-1),q(n,-1); set<int> s; rep(i,n)s.insert(i); if(is[0].size() + is[2].size()>=2){ vector<int> t; rep(i,is[0].size())t.push_back(is[0][i]); rep(i,is[2].size())t.push_back(is[2][i]); while(t.size()>=3)t.pop_back(); p[t[0]] = 0; p[t[1]] = 1; s.erase(0); s.erase(1); rep(i,n){ if(p[i]==-1){ p[i] = *s.begin(); s.erase(s.begin()); } } q = p; swap(q[t[0]],q[t[1]]); rep(i,n){ p[i]++,q[i]++; } Out(p); Out(q); } else if(is[0].size() + is[1].size()>=2){ vector<int> t; rep(i,is[0].size())t.push_back(is[0][i]); rep(i,is[1].size())t.push_back(is[1][i]); while(t.size()>=3)t.pop_back(); p[t[0]] = 0; p[t[1]] = 2; s.erase(0); s.erase(2); rep(i,n){ if(p[i]==-1){ p[i] = *s.begin(); s.erase(s.begin()); } } q = p; swap(q[t[0]],q[t[1]]); rep(i,n){ p[i]++,q[i]++; } Out(p); Out(q); } else{ vector<int> t; rep(i,is[3].size())t.push_back(is[3][i]); while(t.size()>=3)t.pop_back(); p[t[0]] = 1; p[t[1]] = 3; s.erase(1); s.erase(3); rep(i,n){ if(p[i]==-1){ p[i] = *s.begin(); s.erase(s.begin()); } } q = p; swap(q[t[0]],q[t[1]]); rep(i,n){ p[i]++,q[i]++; } Out(p); Out(q); } } return 0; }