結果

問題 No.1706 Many Bus Stops (hard)
コンテスト
ユーザー H3PO4
提出日時 2021-10-08 23:23:49
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 798 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,396 KB
実行使用メモリ 62,752 KB
最終ジャッジ日時 2024-07-23 07:23:13
合計ジャッジ時間 3,885 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 1 WA * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

C, N, M = map(int, input().split())
MOD = 10 ** 9 + 7
iC = pow(C, MOD - 2, MOD)

"""pypy用行列累乗ライブラリは https://github.com/fujihiraryo/alogorithm-python/blob/master/integer/matrix.py から拝借しています。"""


def product(a, b, p=MOD):
    n = len(a)
    c = [[0] * n for i in range(n)]
    for i in range(n):
        for j in range(n):
            c[i][j] = sum([a[i][k] * b[k][j] for k in range(n)]) % p
    return c


def matpow(a, k, p=MOD):
    if k == 1:
        return a
    if k % 2 == 0:
        b = matpow(a, k // 2)
        return product(b, b, p=p)
    return product(a, matpow(a, k - 1), p=p)


A = [[iC, 0, 0, 1 - iC],
     [0, 1 - iC, iC, 0],
     [1, 0, 0, 0],
     [0, 1, 0, 0]]
tmp = matpow(A, N)[0][0]
ans = (1 - pow(1 - tmp, M, MOD)) % MOD
print(ans)
0