結果

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

ソースコード

diff # 

lib C;fun strtoll(s: UInt8*,p: UInt8*,b: Int32): Int64;end
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{|e|C.strtoll(e,nil,10)}
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