結果
| 問題 | No.1097 Remainder Operation |
| コンテスト | |
| ユーザー |
wattaihei
|
| 提出日時 | 2020-06-26 22:53:59 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 114 ms / 2,000 ms |
| コード長 | 468 bytes |
| 記録 | |
| コンパイル時間 | 203 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 89,216 KB |
| 最終ジャッジ日時 | 2024-07-04 23:05:29 |
| 合計ジャッジ時間 | 3,727 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 21 |
ソースコード
import sys
input = sys.stdin.readline
N = int(input())
A = list(map(int, input().split()))
Q = int(input())
Query = [int(input()) for _ in range(Q)]
D = [-1]*N
P = []
x = 0
j = 0
while D[x%N] == -1:
D[x%N] = j
P.append(x)
x = (x + A[x%N])
j += 1
cycle = j - D[x%N]
init = D[x%N]
cyclesum = x - P[D[x%N]]
for K in Query:
if K < init:
ans = P[K]
else:
ans = P[(K-init)%cycle+init] + ((K-init)//cycle)*cyclesum
print(ans)
wattaihei