#974616 (C++17(gcc12)) No.2732 Similar Permutations

提出ソース
結果

問題	No.2732 Similar Permutations
コンテスト
ユーザー	kaichou243
提出日時	2024-04-19 23:22:22
言語	C++17(gcc12) (gcc 12.3.0 + boost 1.87.0)
結果	AC
実行時間	1,245 ms / 2,000 ms
コード長	8,911 bytes
コンパイル時間	13,025 ms
コンパイル使用メモリ	302,472 KB
最終ジャッジ日時	2025-02-21 05:38:49
ジャッジサーバーID （参考情報）	judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン	結果
sample	AC * 2
other	AC * 101
権限があれば一括ダウンロードができます
ソースコード

raw source code
#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;
}
yukicoder

結果

ソースコード
