結果
問題 | No.995 タピオカオイシクナーレ |
ユーザー | convexineq |
提出日時 | 2020-02-21 21:42:17 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 78 ms / 2,000 ms |
コード長 | 1,029 bytes |
コンパイル時間 | 185 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 21,632 KB |
最終ジャッジ日時 | 2024-10-08 22:38:43 |
合計ジャッジ時間 | 2,386 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
# coding: utf-8 # Your code here! import sys readline = sys.stdin.readline read = sys.stdin.read #n,*a = [int(i) for i in read().split()] #a = [int(i) for i in readline().split()] #n = int(input()) MOD=10**9+7 def matmul(A,B,mod): # A,B: 行列 res = [[0]*len(B[0]) for _ in [None]*len(A)] for i, resi in enumerate(res): for k, aik in enumerate(A[i]): for j,bkj in enumerate(B[k]): resi[j] += aik*bkj resi[j] %= mod return res def matpow(A,p,M): #A^p mod M if p%2: return matmul(A, matpow(A,p-1,M), M) elif p > 0: b = matpow(A,p//2,M) return matmul(b,b,M) else: return [[1 if i == j else 0 for j in range(len(A))] for i in range(len(A))] n,m,k,p,q,*a = [int(i) for i in read().split()] naka = sum(a[:m]) soto = sum(a[m:]) MOD = 10**9+7 qinv = pow(q,MOD-2,MOD) qq = (q-p)*qinv%MOD pp = p*qinv%MOD mat = [[qq,pp],[pp,qq]] A = matpow(mat,k,MOD) ans = A[0][0]*naka + A[0][1]*soto print(ans%MOD)