結果

問題 No.1706 Many Bus Stops (hard)
ユーザー Kiri8128
提出日時 2021-10-08 23:21:21
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 707 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,116 KB
実行使用メモリ 62,944 KB
最終ジャッジ日時 2024-07-23 07:17:16
合計ジャッジ時間 3,475 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other WA * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

def mmult(A, B):
    global mod
    n, m, l = len(A), len(B), len(B[0])
    ret = [[0] * l for _ in range(n)]
    for i in range(n):
        for j in range(m):
            for k in range(l):
                ret[i][k] = (ret[i][k] + A[i][j] * B[j][k]) % mod
    return ret
def mpow(A, n):
    if n == 1: return A
    if n == 0: return [[1 if i == j else 0 for j in range(len(A))] for i in range(len(A))]
    return mmult(mpow(A, n - 1), A) if n % 2 else mpow(mmult(A, A), n // 2)

C, N, M = map(int, input().split())
mod = 10 ** 9 + 7
iv = pow(C, mod - 2, mod)
ivv = iv * (C - 2) % mod
X = [[iv, 0, 0, iv], [0, iv, ivv, ivv], [1, 0, 0, 0], [0, 1, 0, 0]]

print((1 - pow(1 - mpow(X, N)[0][0], M, mod)) % mod)
0