結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2021-10-14 08:56:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 717 ms / 2,000 ms |
コード長 | 786 bytes |
コンパイル時間 | 179 ms |
コンパイル使用メモリ | 82,296 KB |
実行使用メモリ | 158,416 KB |
最終ジャッジ日時 | 2024-09-17 16:27:36 |
合計ジャッジ時間 | 10,323 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
import mathN = int(input())A = list(map(int,input().split()))Q = int(input())# 二次元配列作成nex = []nsum = []for _ in range(int(math.log2(10**13))):l1 = [0] * Nl2 = [0] * Nnex.append(l1)nsum.append(l2)# dv[0][0:X]を初期化for n in range(N):nex[0][n] = (n+A[n])%Nnsum[0][n] = A[n]# ダブリングで表を構築for k in range(1, int(math.log2(10**13))):for n in range(N):nex[k][n] = nex[k - 1][nex[k - 1][n]]nsum[k][n] = nsum[k - 1][n] + nsum[k - 1][nex[k - 1][n]]for _ in range(Q):K = int(input())a = []for i in range(int(math.log2(K)) + 1):if K>>i & 1:a.append(i)ans = 0X = 0for i in a:ans += nsum[i][X]X = nex[i][X]print(ans)