結果
| 問題 | 
                            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