結果
問題 | 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 1000000000000000001void 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;}