結果
問題 |
No.1013 〇マス進む
|
ユーザー |
![]() |
提出日時 | 2023-03-31 10:55:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,136 ms / 2,000 ms |
コード長 | 1,760 bytes |
コンパイル時間 | 284 ms |
コンパイル使用メモリ | 82,704 KB |
実行使用メモリ | 256,504 KB |
最終ジャッジ日時 | 2024-09-22 16:09:58 |
合計ジャッジ時間 | 24,129 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 62 |
ソースコード
import sys readline=sys.stdin.readline class Path_Doubling: def __init__(self,N,permutation,lst=None,f=None,e=None): self.N=N self.permutation=permutation self.lst=lst self.f=f self.e=e def Build_Next(self,K): self.K=K self.k=self.K.bit_length() self.permutation_doubling=[[self.permutation[n]] for n in range(self.N)] self.doubling=[[self.lst[n]] for n in range(self.N)] for k in range(1,self.k): for n in range(self.N): self.permutation_doubling[n].append(self.permutation_doubling[self.permutation_doubling[n][k-1]][k-1]) self.doubling[n].append(self.f(self.doubling[n][k-1],self.doubling[self.permutation_doubling[n][k-1]][k-1])) def Permutation_Doubling(self,N,K): for k in range(self.k): if K>>k&1: N=self.permutation_doubling[N][k] return N def Doubling(self,N,K): retu=self.e for k in range(self.k): if K>>k&1: retu=self.f(retu,self.doubling[N][k]) N=self.permutation_doubling[N][k] return N,retu def Bisect(self,x,is_ok): if not is_ok(x): return -1,None K=0 for k in range(self.k-1,-1,-1): if K|1<<k<=self.K and is_ok(self.permutation_doubling[x][k]): K|=1<<k x=self.permutation_doubling[x][k] return K,x N,K=map(int,readline().split()) P=list(map(int,readline().split())) PD=Path_Doubling(N,[(P[i]+i)%N for i in range(N)],[(i+P[i])//N for i in range(N)],lambda x,y:x+y,0) PD.Build_Next(K) for i in range(N): pd=PD.Permutation_Doubling(i,K) _,d=PD.Doubling(i,K) ans=pd+d*N+1 print(ans)