結果
| 問題 |
No.1013 〇マス進む
|
| コンテスト | |
| ユーザー |
tamato
|
| 提出日時 | 2020-03-20 23:49:56 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 553 ms / 2,000 ms |
| コード長 | 634 bytes |
| コンパイル時間 | 421 ms |
| コンパイル使用メモリ | 82,184 KB |
| 実行使用メモリ | 109,668 KB |
| 最終ジャッジ日時 | 2024-12-15 21:07:15 |
| 合計ジャッジ時間 | 14,044 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 62 |
ソースコード
def main():
import sys
input = sys.stdin.readline
N, K = map(int, input().split())
P = list(map(int, input().split()))
LV = K.bit_length()
dist = [[0] * LV for _ in range(N)]
for i in range(N):
dist[i][0] = P[i]
for lv in range(1, LV):
for i in range(N):
j = (i + dist[i][lv-1])%N
dist[i][lv] = dist[i][lv-1] + dist[j][lv-1]
for i in range(N):
ans = i
j = i
for lv in range(LV):
if K >> lv& 1:
ans += dist[j][lv]
j = ans % N
print(ans + 1)
if __name__ == '__main__':
main()
tamato