結果

問題 No.1013 〇マス進む
ユーザー ntuda
提出日時 2025-08-14 22:53:47
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,043 ms / 2,000 ms
コード長 669 bytes
コンパイル時間 321 ms
コンパイル使用メモリ 82,748 KB
実行使用メモリ 359,260 KB
最終ジャッジ日時 2025-08-14 22:54:13
合計ジャッジ時間 23,408 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #

from math import log2
N,K = map(int,input().split())
P = [0] + list(map(int,input().split()))
K2 = int(log2(K)) + 1
X = [[[0,0] for _ in range(N + 1)] for _ in range(K2+1)]

for i in range(1,N+1):
    X[0][i][1] = (P[i] + i - 1) % N + 1
    X[0][i][0] = (P[i] + i - 1) // N
Y = [[0,i] for i in range(N+1)]
i = 0
while K > 0:
    if K & 1:
        for j in range(1, N + 1):
            e,f = Y[j]
            Y[j][1] = X[i][f][1]
            Y[j][0] = e + X[i][f][0]
    i += 1
    for j in range(1,N+1):
        a,b = X[i-1][j]
        c,d = X[i-1][b]
        X[i][j][1] = d
        X[i][j][0] = a + c
    K >>= 1
for i in range(1,N+1):
    print(Y[i][0] * N + Y[i][1])
0