結果
問題 | No.2732 Similar Permutations |
ユーザー |
|
提出日時 | 2024-05-02 09:51:41 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 142 ms / 2,000 ms |
コード長 | 2,762 bytes |
コンパイル時間 | 4,341 ms |
コンパイル使用メモリ | 258,740 KB |
最終ジャッジ日時 | 2025-02-21 10:09:16 |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 101 |
ソースコード
#include <bits/stdc++.h>#include <vector>using namespace std;#include <atcoder/all>using namespace atcoder;using ll = long long;int N;vector<int> A;void solve(){if(N < 5){vector<int> p1(N);iota(p1.begin(),p1.end(),1);do{vector<int> p2(N);iota(p2.begin(),p2.end(),1);do{if(p1 == p2){continue;}int xorsum = 0;for(int i = 0;i<N;i++){xorsum ^= A[i]+p1[i];}for(int i = 0;i<N;i++){xorsum ^= A[i]+p2[i];}if(xorsum == 0){for(int i = 0;i<N;i++){cout << p1[i] << " ";}cout << endl;for(int i = 0;i<N;i++){cout << p2[i] << " ";}cout << endl;return;}}while(next_permutation(p2.begin(),p2.end()));}while(next_permutation(p1.begin(),p1.end()));cout<<-1<<endl;return;}set<int> st;for(int i = 0;i<N;i++){st.insert(i+1);}vector<int> a(N),b(N);vector<int> bin[4];for(int i = 0;i<N;i++){bin[A[i]%4].push_back(i);}if(bin[0].size() > 1){int l = bin[0][0];int r = bin[0][1];a[l] = 1;a[r] = 3;b[r] = 1;b[l] = 3;st.erase(1);st.erase(3);}else if(bin[1].size() >1){int l = bin[1][0];int r = bin[1][1];a[l] = 1;a[r] = 2;b[r] = 1;b[l] = 2;st.erase(1);st.erase(2);}else if(bin[2].size() > 1){int l = bin[2][0];int r = bin[2][1];a[l] = 2;a[r] = 3;b[r] = 2;b[l] = 3;st.erase(2);st.erase(3);}else if(bin[3].size() > 1){int l = bin[3][0];int r = bin[3][1];a[l] = 1;a[r] = 3;b[r] = 1;b[l] = 3;st.erase(1);st.erase(3);}else{cout<<-1<<endl;return;}for(int i = 0;i<N;i++){if(a[i] != 0){continue;}a[i] = *st.begin();b[i] = *st.begin();st.erase(st.begin());}for(int i = 0;i<N;i++){cout << a[i] << " ";}cout << endl;for(int i = 0;i<N;i++){cout << b[i] << " ";}cout << endl;}signed main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin >> N;A = vector<int>(N);for (int i = 0; i < N; i++) {cin >> A[i];}solve();}