結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2020-06-26 22:32:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 168 ms / 2,000 ms |
コード長 | 837 bytes |
コンパイル時間 | 145 ms |
コンパイル使用メモリ | 81,984 KB |
実行使用メモリ | 108,204 KB |
最終ジャッジ日時 | 2024-07-04 22:16:35 |
合計ジャッジ時間 | 4,557 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
n = int(input()) a = list(map(int, input().split())) q = int(input()) k = [int(input()) for i in range(q)] ans = [0] ind = [-1] memo = {} for i in range(3 * n): tmp = ans[-1] r = tmp % n if r in memo: lap_length = len(ans) - memo[r] lap_diff = ans[-1] - ans[memo[r] - 1] init_length = len(ans) - lap_length - 1 break memo[r] = len(ans) ans.append(tmp + a[r]) ind.append(r) ru = [0] * (lap_length + 1) for i in range(lap_length): ru[i + 1] = ru[i] + ans[memo[r] + i] - ans[memo[r] - 1 + i] #print(ans) #print(ind) #print(init_length, lap_length, lap_diff) #print(ru) for num in k: if num < len(ans): print(ans[num]) else: res = ans[init_length] laps, rem = divmod(num - init_length, lap_length) print(res + laps * lap_diff + ru[rem])