結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
コンテスト
ユーザー r6eve
提出日時 2017-08-17 16:43:23
言語 OCaml
(5.2.1)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 799 bytes
コンパイル時間 297 ms
コンパイル使用メモリ 21,320 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-09 00:25:41
合計ジャッジ時間 965 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

let mul a b m =
  let na0 = Array.length a in
  let nb0 = Array.length b in
  let nb1 = Array.length b.(0) in
  let c = Array.make_matrix na0 nb1 0 in
  for i = 0 to na0 - 1 do
    for k = 0 to nb0 - 1 do
      for j = 0 to nb1 - 1 do
        c.(i).(j) <- (c.(i).(j) + a.(i).(k) * b.(k).(j)) mod m
      done
    done
  done;
  c


let fib (n, m) =
  let a = Array.make_matrix 2 2 0 in
  a.(0).(0) <- 1; a.(0).(1) <- 1;
  a.(1).(0) <- 1; a.(1).(1) <- 0;
  let b = Array.make_matrix 2 2 0 in
  b.(0).(0) <- 1; b.(0).(1) <- 0;
  b.(1).(0) <- 0; b.(1).(1) <- 1;
  let rec doit n a b =
    if n = 0 then b
    else doit (n / 2) (mul a a m) (if n mod 2 = 0 then b else mul b a m) in
  let c = doit (n - 1) a b in
  c.(1).(0)

let () = Scanf.scanf "%d %d " (fun n m -> n, m) |> fib |> Printf.printf "%d\n"
0