結果

問題 No.1073 無限すごろく
ユーザー ntuda
提出日時 2025-01-06 21:46:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 554 bytes
コンパイル時間 1,554 ms
コンパイル使用メモリ 81,852 KB
実行使用メモリ 62,396 KB
最終ジャッジ日時 2025-01-06 21:46:50
合計ジャッジ時間 3,399 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

MOD = 10 ** 9 + 7
N = int(input())
A = [[0] * 6 for _ in range(6)]
X = [[0] * 6 for _ in range(6)]
for i in range(6):
    A[0][i] = pow(6, -1, MOD)
    X[i][i] = 1
for i in range(5):
    A[i + 1][i] = 1
while N > 0:
    if N & 1:
        X = mt(A, X)
    A = mt(A, A)
    N >>= 1
print(X[0][0])
0