結果

問題 No.1097 Remainder Operation
ユーザー c-yanc-yan
提出日時 2020-06-26 23:16:08
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 373 ms / 2,000 ms
コード長 741 bytes
コンパイル時間 230 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 25,144 KB
最終ジャッジ日時 2024-07-04 23:52:09
合計ジャッジ時間 6,748 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import stdin
readline = stdin.readline

N = int(readline())
A = list(map(int, readline().split()))

X = 0
b = [-1]
c = [X]
used = set()
while True:
    r = X % N
    X += A[r]
    if r in used:
        break
    used.add(r)
    b.append(r)
    c.append(X)
b.append(r)
c.append(X)
loop_start = b.index(b[-1])
loop_len = len(b) - loop_start - 1
#print(loop_start)
#print(loop_len)
#print(b)
#print(c)

Q = int(readline())
for _ in range(Q):
    K = int(readline())
    if K < len(b):
        print(c[K])
    else:
        K -= loop_start
        a = K // loop_len
        K %= loop_len
        K += loop_start
        #print(a)
        #print(K)
        #print(c[-1] - c[loop_start])
        print(c[K] + a * (c[-1] - c[loop_start]))
0