結果

問題 No.2732 Similar Permutations
ユーザー kaichou243
提出日時 2024-04-19 23:00:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,330 bytes
コンパイル時間 2,949 ms
コンパイル使用メモリ 215,052 KB
最終ジャッジ日時 2025-02-21 05:15:54
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 89 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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);
	for(int i=0;i<n;i++) cin>>a[i];
	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;
	}
	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;
	}
	cout<<-1<<endl;
}
0