結果
問題 | No.995 タピオカオイシクナーレ |
ユーザー |
![]() |
提出日時 | 2020-02-21 21:50:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 103 ms / 2,000 ms |
コード長 | 895 bytes |
コンパイル時間 | 436 ms |
コンパイル使用メモリ | 82,464 KB |
実行使用メモリ | 77,820 KB |
最終ジャッジ日時 | 2024-10-08 22:41:33 |
合計ジャッジ時間 | 3,572 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
def _mul(A, B, MOD): C = [[0] * len(B[0]) for i in range(len(A))] for i in range(len(A)): for k in range(len(B)): for j in range(len(B[0])): C[i][j] = (C[i][j] + A[i][k]*B[k][j]) % MOD return C def pow_matrix(A, n, MOD): """A**nをダブリングによって求める""" B = [[0] * len(A) for i in range(len(A))] for i in range(len(A)): B[i][i] = 1 while n > 0: if n & 1: B = _mul(A, B, MOD) A = _mul(A, A, MOD) n = n // 2 return B n, m, k, p, q = list(map(int, input().split())) b = [int(input()) for i in range(n)] MOD = 10**9 + 7 milk = sum(b[0:m]) kitchen = sum(b[m:n]) pq = p * pow(q, MOD - 2, MOD) % MOD qpq = (q-p) * pow(q, MOD - 2, MOD) % MOD A = [[qpq, pq], [pq, qpq]] matrix = pow_matrix(A, k, MOD) ans = milk * matrix[0][0] + kitchen * matrix[0][1] print(ans % MOD)