結果
問題 | No.1097 Remainder Operation |
ユーザー |
![]() |
提出日時 | 2020-11-21 00:37:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 431 ms / 2,000 ms |
コード長 | 1,390 bytes |
コンパイル時間 | 1,642 ms |
コンパイル使用メモリ | 82,452 KB |
実行使用メモリ | 153,440 KB |
最終ジャッジ日時 | 2024-07-23 14:11:54 |
合計ジャッジ時間 | 6,704 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
import sysdef input(): return sys.stdin.readline().strip()def list2d(a, b, c): return [[c] * b for i in range(a)]def list3d(a, b, c, d): return [[[d] * c for k in range(b)] for i in range(a)]def list4d(a, b, c, d, e): return [[[[e] * d for k in range(c)] for k in range(b)] for i in range(a)]def ceil(x, y=1): return int(-(-x // y))def INT(): return int(input())def MAP(): return map(int, input().split())def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)]def Yes(): print('Yes')def No(): print('No')def YES(): print('YES')def NO(): print('NO')sys.setrecursionlimit(10**9)INF = 10**19MOD = 10**9 + 7EPS = 10**-10def doubling(MAX, A):""" ダブリング """N = len(A)nxt = list2d(MAX, N, -1)sm = list2d(MAX, N, 0)for i, a in enumerate(A):nxt[0][i] = (i+a) % Nsm[0][i] = afor k in range(1, MAX):for i in range(N):nxt[k][i] = nxt[k-1][nxt[k-1][i]]sm[k][i] = sm[k-1][i] + sm[k-1][nxt[k-1][i]]return (nxt, sm)N = INT()A = LIST()MX = 40nxt, sm = doubling(MX, A)# x = cur = 0# for i in range(10):# x += sm[0][cur]# cur = nxt[0][cur] % N# print(x, cur)for _ in range(INT()):k = INT()x = cur = 0for i in range(MX-1, -1, -1):if k>>i & 1:x += sm[i][cur]cur = nxt[i][cur] % Nprint(x)