結果

問題 No.995 タピオカオイシクナーレ
コンテスト
ユーザー ntuda
提出日時 2025-01-11 09:53:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 155 ms / 2,000 ms
コード長 699 bytes
コンパイル時間 706 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 76,920 KB
最終ジャッジ日時 2025-01-11 09:53:59
合計ジャッジ時間 3,978 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10 ** 9 + 7
D = 2
def mt(A, B):
    C = [[0] * D for _ in range(D)]
    for i in range(D):
        for j in range(D):
            tmp = 0
            for k in range(D):
                tmp += A[i][k] * B[k][j]
                tmp %= MOD
            C[i][j] = tmp
    return C

N, M, K, P, Q = map(int, input().split())
B = [int(input()) for _ in range(N)]
a = 0
b = 0
for i in range(N):
    if i < M:
        a += B[i]
    else:
        b += B[i]


move = (P * pow(Q, -1, MOD)) % MOD
notmove = (1 - move) % MOD

A = [[notmove,move],[move,notmove]]
X = [[1,0],[0,1]]
while K > 0:
    if K & 1:
        X = mt(A,X)
    A = mt(A,A)
    K >>= 1
ans = (X[0][0] * a + X[0][1] * b) % MOD
print(ans)
0