結果

問題 No.718 行列のできるフィボナッチ数列道場 (1)
ユーザー はむ吉🐹
提出日時 2018-07-29 14:58:58
言語 Ruby
(3.4.1)
結果
AC  
実行時間 101 ms / 2,000 ms
コード長 545 bytes
コンパイル時間 74 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 13,312 KB
最終ジャッジ日時 2024-07-17 22:55:21
合計ジャッジ時間 3,321 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

#!/usr/bin/env ruby


require "matrix"


M = 1000000007


def mod_mul(a, b, mod=M)
  return (a * b).map{|x| x % mod}
end


def mod_pow(a, n, mod=M)
  raise "assert" unless a.square?
  b = Matrix.I(a.column_size)
  while n > 0
    if n % 2 == 1
      b = mod_mul(b, a, mod)
    end
    a = mod_mul(a, a, mod)
    n /= 2
  end
  return b
end


def solve(n, mod=M)
  a = Matrix[[1, 1], [1, 0]]
  res = mod_mul(mod_pow(a, n, mod), Vector[1, 0])
  return res[0] * res[1] % mod
end


def main()
  n = gets.to_i
  ans = solve(n)
  puts(ans)
end


main
0