結果

問題 No.1073 無限すごろく
ユーザー 👑 Kazun
提出日時 2020-06-05 22:55:07
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 708 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,204 KB
実行使用メモリ 128,112 KB
最終ジャッジ日時 2024-12-17 17:13:41
合計ジャッジ時間 32,213 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

def m(A,B,Y):
    C=[]
    for i in range(6):
        D=[]
        for j in range(6):
            S=0
            for k in range(6):
                S=(S+A[i][k]*B[k][j])%Y

            D.append(S)
        C.append(D)
    return C

Y=10**9+7
a=166666668

N=int(input())

if N<=6:
    print((7**(N-1)*a**N)%Y)
    sys.exit()

P=[[a,a,a,a,a,a],[1,0,0,0,0,0],[0,1,0,0,0,0],[0,0,1,0,0,0],[0,0,0,1,0,0],[0,0,0,0,1,0]]
Q=[P]
k=2
while k<=N:
    R=Q[-1]
    k*=2
    Q.append(m(R,R,Y))

M=N-6
Z=[[1*(i==j) for j in range(6)] for i in range(6)]
for i in range(k):
    if M%2==1:
        Z=m(Z,Q[i],Y)
    M=M>>1

W=0
L=a
for j in range(6):
    W=(W+Z[0][5-j]*L)%Y
    L=(L*a*7)%Y

print(W)
0