結果

問題 No.1013 〇マス進む
ユーザー titia
提出日時 2020-03-21 06:49:17
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 584 bytes
コンパイル時間 144 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 284,320 KB
最終ジャッジ日時 2024-12-17 18:14:08
合計ジャッジ時間 91,534 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 41 TLE * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

N,K=map(int,input().split())
P=list(map(int,input().split()))
mod=N

P2=P[-1:]+P[:-1]

NEXT=[[0]*N for i in range(31)]
DIS=[[0]*N for i in range(31)]

for i in range(N):
    DIS[0][i]=P2[i]
    NEXT[0][i]=(i+DIS[0][i])%mod

for j in range(1,31):
    for i in range(N):
        DIS[j][i]=DIS[j-1][i]+DIS[j-1][NEXT[j-1][i]]
        NEXT[j][i]=(i+DIS[j][i])%mod

A=[]
for i in range(N):
    NOW=i
    ANS=0
    U=K
    while U:
        b=U.bit_length()-1
        ANS+=DIS[b][NOW]
        NOW=NEXT[b][NOW]
        U-=(1<<b)

    A.append(ANS+i)

for a in A[1:]:
    print(a)
print(A[0]+N)
0