結果

問題 No.1073 無限すごろく
ユーザー 👑 Kazun
提出日時 2020-06-05 23:09:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 751 bytes
コンパイル時間 253 ms
コンパイル使用メモリ 82,512 KB
実行使用メモリ 66,112 KB
最終ジャッジ日時 2024-12-17 17:46:04
合計ジャッジ時間 2,859 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
from math import log
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]

for _ in range(int(log(N,2))+1):
    R=Q[-1]
    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(int(log(N,2))+1):
    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