結果
問題 |
No.891 隣接3項間の漸化式
|
ユーザー |
|
提出日時 | 2019-11-25 23:32:33 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 484 ms / 2,000 ms |
コード長 | 711 bytes |
コンパイル時間 | 181 ms |
コンパイル使用メモリ | 12,416 KB |
実行使用メモリ | 44,492 KB |
最終ジャッジ日時 | 2024-10-15 00:19:26 |
合計ジャッジ時間 | 23,716 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
import numpy as np a,b,n=map(int,input().split()) m=10**9+7 k=n//2 def dot(A,B): n=A.shape[0] ans=np.zeros(A.shape,dtype=np.int64) A=A%m B=B%m for i in range(n): for j in range(n): for k in range(n): ans[i,j]=ans[i,j]+(A[i,k]*B[k,j])%m return ans def pow(A,k): last=np.eye(A.shape[0]) cur_A=A while True: if k==0: cur_A=np.eye(A.shape[0]) break elif k%2==0: cur_A=dot(cur_A,cur_A) k=k/2 else: k=k-1 last=dot(last,cur_A) return dot(cur_A,last) l=1-(n-2*k) A=np.array([[a*a+b,a],[a*b,b]]) B=pow(A,k) print(int(B[0,l]))