結果

問題 No.1097 Remainder Operation
ユーザー flippergoflippergo
提出日時 2024-10-14 14:12:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 375 ms / 2,000 ms
コード長 791 bytes
コンパイル時間 754 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 111,872 KB
最終ジャッジ日時 2024-10-14 14:12:12
合計ジャッジ時間 8,016 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 53 ms
62,976 KB
testcase_01 AC 54 ms
63,360 KB
testcase_02 AC 54 ms
63,616 KB
testcase_03 AC 57 ms
63,360 KB
testcase_04 AC 58 ms
63,616 KB
testcase_05 AC 56 ms
63,232 KB
testcase_06 AC 64 ms
63,232 KB
testcase_07 AC 123 ms
77,440 KB
testcase_08 AC 115 ms
77,696 KB
testcase_09 AC 133 ms
77,952 KB
testcase_10 AC 117 ms
77,568 KB
testcase_11 AC 119 ms
77,440 KB
testcase_12 AC 305 ms
90,240 KB
testcase_13 AC 306 ms
90,752 KB
testcase_14 AC 328 ms
90,368 KB
testcase_15 AC 316 ms
90,368 KB
testcase_16 AC 305 ms
90,368 KB
testcase_17 AC 327 ms
90,880 KB
testcase_18 AC 375 ms
111,616 KB
testcase_19 AC 333 ms
111,744 KB
testcase_20 AC 369 ms
111,616 KB
testcase_21 AC 327 ms
111,744 KB
testcase_22 AC 329 ms
111,872 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from ast import Name
N = int(input())
A = list(map(int,input().split()))
B = [0]
setB = {0}
cur = 0
while True:
    ncur = (cur+A[cur%N])%N
    if ncur in setB:break
    setB.add(ncur)
    B.append(ncur)
    cur = ncur
ind = B.index(ncur)
B0 = B[:ind]
B1 = B[ind:]
S = len(B0)
if S>0:
    cumA0 = [0]*S
    cumA0[0] = A[B0[0]]
    for i in range(1,S):
        cumA0[i] = cumA0[i-1]+A[B0[i]]
T = len(B1)
cumA1 = [0]*T
cumA1[0] = A[B1[0]]
for i in range(1,T):
    cumA1[i] = cumA1[i-1]+A[B1[i]]
Q = int(input())
for _ in range(Q):
    K = int(input())
    ans = 0
    if K<=S:
        ans = cumA0[K-1]
    else:
        if S>0:
            ans = cumA0[S-1]
        K -= S
        q = K//T
        r = K%T
        ans += q*cumA1[T-1]
        if r>0:
            ans += cumA1[r-1]
    print(ans)
0