結果

問題 No.1704 Many Bus Stops (easy)
コンテスト
ユーザー Kiri8128
提出日時 2021-10-08 22:54:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 578 ms / 2,000 ms
コード長 728 bytes
コンパイル時間 414 ms
コンパイル使用メモリ 82,184 KB
実行使用メモリ 79,336 KB
最終ジャッジ日時 2024-07-23 06:05:38
合計ジャッジ時間 17,843 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
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)

mod = 10 ** 9 + 7
i3 = (mod + 1) // 3
X = [[i3, 0, 0, 0, i3, i3], [0, i3, 0, i3, 0, i3], [0, 0, i3, i3, i3, 0], [1, 0, 0, 0, 0, 0], [0, 1, 0, 0, 0, 0], [0, 0, 1, 0, 0, 0]]

T = int(input())
for _ in range(T):
    print(mpow(X, int(input()))[0][0])
0