結果

問題 No.1706 Many Bus Stops (hard)
ユーザー Kiri8128
提出日時 2021-10-08 23:22:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 49 ms / 2,000 ms
コード長 732 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 62,544 KB
最終ジャッジ日時 2024-07-23 07:18:49
合計ジャッジ時間 3,430 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 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)
iv1 = iv * (C - 1) % mod
iv2 = iv * (C - 2) % mod
X = [[iv, 0, 0, iv], [0, iv, iv1, iv2], [1, 0, 0, 0], [0, 1, 0, 0]]

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