結果

問題 No.1097 Remainder Operation
ユーザー flippergo
提出日時 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
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

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)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0