結果
| 問題 |
No.995 タピオカオイシクナーレ
|
| コンテスト | |
| ユーザー |
👑 SPD_9X2
|
| 提出日時 | 2025-08-09 23:42:53 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 1,229 bytes |
| コンパイル時間 | 347 ms |
| コンパイル使用メモリ | 82,436 KB |
| 実行使用メモリ | 78,268 KB |
| 最終ジャッジ日時 | 2025-08-09 23:42:56 |
| 合計ジャッジ時間 | 3,449 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 10 WA * 13 |
ソースコード
import sys
from sys import stdin
#行列A,Bの積(不可能だとエラー,modに0以下の値を指定するとmodを取らなくなる)
def matrix_mul(A,B,mod):
ans = [ [0] * len(B[0]) for i in range(len(A)) ]
for ai in range(len(A)):
for bj in range(len(B[0])):
now = 0
for same in range(len(A[0])):
now += A[ai][same] * B[same][bj]
if mod > 0:
ans[ai][bj] = now % mod
else:
ans[ai][bj] = now
return ans
#行列Aのx乗(当然正方行列じゃないとだめ)
def matrix_pow(A,x,mod):
B = [[A[i][j] for j in range(len(A[0]))] for i in range(len(A))]
ans = [[0] * len(A[0]) for i in range(len(A))]
for i in range(len(A)):
ans[i][i] = 1
while x > 0:
if x % 2 == 1:
ans = matrix_mul(ans,B,mod)
B = matrix_mul(B,B,mod)
x//=2
return ans
mod = 10**9+7
N,M,K,p,q = map(int,input().split())
b = [int(input()) for i in range(N)]
div = p * pow(q,mod-2,mod) % mod
a = [ [ sum(b[:M]) , sum(b[M:]) ] ]
H = [
[div , (1-div)%mod],
[(1-div)%mod, div],
]
ans = matrix_mul(a,matrix_pow(H,K,mod),mod)
print (ans[0][0] % mod)
SPD_9X2