結果

問題 No.720 行列のできるフィボナッチ数列道場 (2)
コンテスト
ユーザー ciel
提出日時 2020-02-08 22:26:56
言語 Crystal
(1.14.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 467 bytes
コンパイル時間 12,375 ms
コンパイル使用メモリ 299,732 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-30 20:01:02
合計ジャッジ時間 12,949 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

M=1000000007
A=[[1_i64,1_i64],[1_i64,0_i64]]
def mul(a,b)
	r=(0...a.size).map{[0_i64]*b[0].size}
	a.size.times{|y|
		b[0].size.times{|x|
			r[y][x]=b.size.times.reduce(0_i64){|s,i|(s+a[y][i]*b[i][x])%M}
		}
	}
	r
end
def pow(a,n)
	e=(0...a.size).map{|i|[0]*i+[1]+[0]*(a.size-1-i)}
	while n>0
		e=mul(a,e) if n%2>0
		a=mul(a,a)
		n//=2
	end
	e[-1][0]
end
n,m=gets.not_nil!.split.map &.to_i64
a=pow(A,m+1)
b=pow(A,m)
c=pow(A,m-1)
puts pow([[a,b,0],[b,c,0],[0,1,1]],n+1)
0