結果
問題 |
No.995 タピオカオイシクナーレ
|
ユーザー |
👑 |
提出日時 | 2022-07-05 21:25:33 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 135 ms / 2,000 ms |
コード長 | 844 bytes |
コンパイル時間 | 227 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 76,672 KB |
最終ジャッジ日時 | 2024-12-16 07:43:41 |
合計ジャッジ時間 | 4,163 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
MOD = 10 ** 9 + 7 def matpow(A, B, w): l = len(A) while w: if w & 1: C = [0] * l for i in range(l): for j in range(l): C[i] += A[i][j] * B[j] C[i] %= MOD B = C C = [[0] * l for _ in range(l)] for i in range(l): for j in range(l): for k in range(l): C[i][j] += A[i][k] * A[k][j] C[i][j] %= MOD A = C w >>= 1 return B n, m, k, p, q = map(int, input().split()) B = [0, 0] for _ in range(m): B[0] += int(input()) B[0] %= MOD for _ in range(n - m): B[1] += int(input()) B[1] %= MOD A = [[0, 0], [0, 0]] A[0][1] = A[1][0] = p * pow(q, MOD - 2, MOD) % MOD A[0][0] = A[1][1] = (1 - A[0][1]) % MOD print(matpow(A, B, k)[0])