結果

問題 No.720 行列のできるフィボナッチ数列道場 (2)
コンテスト
ユーザー ciel
提出日時 2018-07-28 00:42:19
言語 Crystal
(1.14.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 466 bytes
コンパイル時間 14,608 ms
コンパイル使用メモリ 300,632 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-30 19:34:26
合計ジャッジ時間 13,105 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 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