結果

問題 No.1500 Super Knight
ユーザー googol_S0
提出日時 2021-05-07 21:34:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 688 bytes
コンパイル時間 300 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 61,056 KB
最終ジャッジ日時 2024-09-15 09:27:32
合計ジャッジ時間 2,969 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

N=int(input())
mod=10**9+7
def m(a,b):
  r=[[0]*len(b[0]) for i in range(len(a))]
  for i in range(len(a)):
    for k in range(len(b)):
      for j in range(len(b[0])):
        r[i][j]=(r[i][j]+a[i][k]*b[k][j])%mod
  return r

def p(a,n):
  r=[[0]*len(a) for i in range(len(a))]
  b=[]
  for i in range(len(a)):
    r[i][i]=1
    b.append(a[i][:])
  l=n
  while l>0:
    if l&1:
      r=m(b,r)
    b=m(b,b)
    l>>=1
  return r

A=[[0]*6 for i in range(6)]
for i in range(5):
  A[i+1][i]=1
A[0][2]=1
A[0][1]=-3
A[0][0]=3
B=[[0]*6 for i in range(6)]
B[5][-1]=1
B[4][-1]=12
B[3][-1]=65
B[2][-1]=172
B[1][-1]=297
B[0][-1]=456
if N<=5:
  print(B[5-N][-1])
  exit()
print(m(p(A,N-5),B)[0][-1])
0