結果
問題 | No.2732 Similar Permutations |
ユーザー |
![]() |
提出日時 | 2024-04-21 01:52:50 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,214 bytes |
コンパイル時間 | 91 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 46,732 KB |
最終ジャッジ日時 | 2024-10-13 00:20:40 |
合計ジャッジ時間 | 40,995 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 73 WA * 28 |
ソースコード
import sysinput = sys.stdin.readlineN=int(input())A=list(map(int,input().split()))if N==2:x=(A[0]+1)^(A[1]+2)y=(A[0]+2)^(A[1]+1)if x==y:print(1,2)print(2,1)else:print(-1)exit()X=[[] for i in range(4)]for i in range(N):X[A[i]%4].append(i)for i in range(4):if i>=2 and len(X[i])>=2:k=X[i][0]l=X[i][1]ANS1=[0]*NANS2=[0]*NANS1[k]=2ANS1[l]=3ANS2[k]=3ANS2[l]=2LL=[2,3]now=1for i in range(N):if ANS1[i]==0:ANS1[i]=nownow+=1while now in LL:now+=1now=1for i in range(N):if ANS2[i]==0:ANS2[i]=nownow+=1while now in LL:now+=1print(*ANS1)print(*ANS2)exit()if i<=1 and len(X[i])>=2:k=X[i][0]l=X[i][1]ANS1=[0]*NANS2=[0]*NANS1[k]=2ANS1[l]=1ANS2[k]=1ANS2[l]=2LL=[1,2]now=1for i in range(N):if ANS1[i]==0:ANS1[i]=nownow+=1while now in LL:now+=1now=1for i in range(N):if ANS2[i]==0:ANS2[i]=nownow+=1while now in LL:now+=1print(*ANS1)print(*ANS2)exit()from itertools import permutationsfrom collections import Counterif N<=5:L=list(permutations(range(N)))ANS=[0]*len(L)for i in range(len(L)):xor=0for j in range(N):xor^=(A[j]+L[i][j]+1)ANS[i]=xorC=Counter(ANS)ans=-1for c in C:if C[c]>=2:ans=cbreakif ans==-1:print(ans)exit()IND=[]for i in range(len(L)):if ANS[i]==ans:IND.append(i)AA=[0]*Nfor i in range(N):AA[i]=L[IND[0]][i]+1print(*AA)AA=[0]*Nfor i in range(N):AA[i]=L[IND[1]][i]+1print(*AA)