結果
問題 | No.283 スライドパズルと魔方陣 |
ユーザー |
![]() |
提出日時 | 2023-07-14 14:35:49 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 8,183 bytes |
コンパイル時間 | 244 ms |
コンパイル使用メモリ | 13,568 KB |
実行使用メモリ | 13,184 KB |
最終ジャッジ日時 | 2024-09-15 23:36:52 |
合計ジャッジ時間 | 6,550 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 25 WA * 14 |
ソースコード
import bisectimport copyimport decimalimport fractionsimport heapqimport itertoolsimport mathimport randomimport sysimport timefrom collections import Counter,deque,defaultdictfrom functools import lru_cache,reducefrom heapq import heappush,heappop,heapify,heappushpop,_heappop_max,_heapify_maxdef _heappush_max(heap,item):heap.append(item)heapq._siftdown_max(heap, 0, len(heap)-1)def _heappushpop_max(heap, item):if heap and item < heap[0]:item, heap[0] = heap[0], itemheapq._siftup_max(heap, 0)return itemfrom math import gcd as GCDread=sys.stdin.readreadline=sys.stdin.readlinereadlines=sys.stdin.readlineswrite=sys.stdout.writedef Magic_Circle(N):if N==2:magic_cycle=Noneelif N%4==0:magic_cycle=[[None]*N for x in range(N)]for x in range(N):for y in range(N):if (x-y)%4==0 or (x+y)%4==3:magic_cycle[x][y]=x*N+yfor x in range(N-1,-1,-1):for y in range(N-1,-1,-1):if magic_cycle[x][y]==None:magic_cycle[x][y]=(N-1-x)*N+(N-1-y)elif N%4==2:n=N//2magic_cycle=[[None]*N for x in range(N)]mc=Magic_Circle(n)L=[[3,0],[1,2]]U=[[0,3],[1,2]]X=[[0,3],[2,1]]for x in range(n):for y in range(n):if n//2+1<x:LUX=Xelif x==n//2+1 and y!=n//2 or (x,y)==(n//2,n//2):LUX=Uelse:LUX=Lfor i in range(2):for j in range(2):magic_cycle[x*2+i][y*2+j]=mc[x][y]*4+LUX[i][j]else:n=N//2magic_cycle=[[None]*N for i in range(N)]for i in range(N):for j in range(N):magic_cycle[(i+j-n)%N][(-i+j+n)%N]=i*N+jreturn magic_cycleclass Segment_Tree:def __init__(self,N,f,e,lst=None,dynamic=False):self.f=fself.e=eself.N=Nif dynamic:self.segment_tree=defaultdict(lambda:self.e)else:if lst==None:self.segment_tree=[self.e]*2*self.Nelse:assert len(lst)<=self.Nself.segment_tree=[self.e]*self.N+[x for x in lst]+[self.e]*(N-len(lst))for i in range(self.N-1,0,-1):self.segment_tree[i]=self.f(self.segment_tree[i<<1],self.segment_tree[i<<1|1])def __getitem__(self,i):if type(i)==int:if -self.N<=i<0:return self.segment_tree[i+self.N*2]elif 0<=i<self.N:return self.segment_tree[i+self.N]else:raise IndexError("list index out of range")else:a,b,c=i.start,i.stop,i.stepif a==None:a=self.Nelse:a+=self.Nif b==None:b=self.N*2else:b+=self.Nreturn self.segment_tree[slice(a,b,c)]def __setitem__(self,i,x):if -self.N<=i<0:i+=self.N*2elif 0<=i<self.N:i+=self.Nelse:raise IndexError("list index out of range")self.segment_tree[i]=xwhile i>1:i>>= 1self.segment_tree[i]=self.f(self.segment_tree[i<<1],self.segment_tree[i<<1|1])def Build(self,lst):for i,x in enumerate(lst,self.N):self.segment_tree[i]=xfor i in range(self.N-1,0,-1):self.segment_tree[i]=self.f(self.segment_tree[i<<1],self.segment_tree[i<<1|1])def Fold(self,L=None,R=None):if L==None:L=self.Nelse:L+=self.Nif R==None:R=self.N*2else:R+=self.NvL=self.evR=self.ewhile L<R:if L&1:vL=self.f(vL,self.segment_tree[L])L+=1if R&1:R-=1vR=self.f(self.segment_tree[R],vR)L>>=1R>>=1return self.f(vL,vR)def Fold_Index(self,L=None,R=None):if L==None:L=self.Nelse:L+=self.Nif R==None:R=self.N*2else:R+=self.Nif L==R:return Nonex=self.Fold(L-self.N,R-self.N)while L<R:if L&1:if self.segment_tree[L]==x:i=LbreakL+=1if R&1:R-=1if self.segment_tree[R]==x:i=RbreakL>>=1R>>=1while i<self.N:if self.segment_tree[i]==self.segment_tree[i<<1]:i<<=1else:i<<=1i|=1i-=self.Nreturn idef Bisect_Right(self,L=None,f=None):if L==self.N:return self.Nif L==None:L=0L+=self.Nvl=self.evr=self.el,r=L,self.N*2while l<r:if l&1:vl=self.f(vl,self.segment_tree[l])l+=1if r&1:r-=1vr=self.f(self.segment_tree[r],vr)l>>=1r>>=1if f(self.f(vl,vr)):return self.Nv=self.ewhile True:while L%2==0:L>>=1vv=self.f(v,self.segment_tree[L])if f(vv):v=vvL+=1else:while L<self.N:L<<=1vv=self.f(v,self.segment_tree[L])if f(vv):v=vvL+=1return L-self.Ndef Bisect_Left(self,R=None,f=None):if R==0:return 0if R==None:R=self.NR+=self.Nvl=self.evr=self.el,r=self.N,Rwhile l<r:if l&1:vl=self.f(vl,self.segment_tree[l])l+=1if r&1:r-=1vr=self.f(self.segment_tree[r],vr)l>>=1r>>=1if f(self.f(vl,vr)):return 0v=self.ewhile True:R-=1while R>1 and R%2:R>>=1vv=self.f(self.segment_tree[R],v)if f(vv):v=vvelse:while R<self.N:R=2*R+1vv=self.f(self.segment_tree[R],v)if f(vv):v=vvR-=1return R+1-self.Ndef __str__(self):return "["+", ".join(map(str,self.segment_tree[self.N:]))+"]"def Inversion_Number(lst,weight=False,weakly=False):compress,decompress=Compress(lst)compressed_lst=[compress[x] for x in lst]N=len(compress)if not weight:weight=[1]*len(lst)ST=Segment_Tree(N,lambda x,y:x+y,0)inversion_number=0for c,x in zip(weight,compressed_lst):inversion_number+=ST.Fold(x if weakly else x+1,N)*cST[x]+=creturn inversion_numberdef Compress(lst):decomp=sorted(list(set(lst)))comp={x:i for i,x in enumerate(decomp)}return comp,decompN=int(readline())if N==2:print("impossible")exit()bl=0A=[list(map(int,readline().split())) for i in range(N)]for i in range(N):for j in range(N):A[i][j]=(A[i][j]-1)%(N*N)+1a=[]for i in range(N):a+=A[i]for i in range(N):for j in range(N):if A[i][j]==N*N:bl^=(i+j)%2MC=Magic_Circle(N)mc=[]for i in range(N):mc+=MC[i]for i in range(N):for j in range(N):if MC[i][j]==0:bl^=(i+j)%2if bl^Inversion_Number(a)%2^Inversion_Number(mc)%2:if N%2==0:MC=MC[::-1]else:MC[0],MC[N-1]=MC[N-1],MC[0]print("possible")for i in range(N):print(*[x+1 for x in MC[i]])