結果

問題 No.3 ビットすごろく
ユーザー kashiwagi513kashiwagi513
提出日時 2019-03-01 12:24:16
言語 OCaml
(5.1.0)
結果
AC  
実行時間 4 ms / 5,000 ms
コード長 1,417 bytes
コンパイル時間 246 ms
コンパイル使用メモリ 19,916 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-17 08:45:29
合計ジャッジ時間 1,271 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 4 ms
5,376 KB
testcase_11 AC 3 ms
5,376 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 3 ms
5,376 KB
testcase_15 AC 3 ms
5,376 KB
testcase_16 AC 4 ms
5,376 KB
testcase_17 AC 4 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 4 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 3 ms
5,376 KB
testcase_23 AC 3 ms
5,376 KB
testcase_24 AC 4 ms
5,376 KB
testcase_25 AC 4 ms
5,376 KB
testcase_26 AC 1 ms
5,376 KB
testcase_27 AC 1 ms
5,376 KB
testcase_28 AC 4 ms
5,376 KB
testcase_29 AC 4 ms
5,376 KB
testcase_30 AC 1 ms
5,376 KB
testcase_31 AC 1 ms
5,376 KB
testcase_32 AC 3 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

let get arr i   = arr.(i-1)
let set arr i v = arr.(i-1) <- v

let rec value_of_cell n acc =
  if n = 0 then acc
  else value_of_cell (n / 2) (acc + (n mod 2))

let value_of_cell n = value_of_cell n 0

let gen_graph n =
  let g = Array.make n [] in
  for i = 1 to n do
    let v = value_of_cell i in
    if i+v <= n then set g (i+v) (i::(get g (i+v)));
    if 1 <= i-v then set g (i-v) (i::(get g (i-v)))
  done; g

let print_list pr ls = 
  let rec inner = function
  | []    -> ()
  | l::[] -> pr l
  | l::ls -> (pr l); print_string ", "; inner ls
  in print_string "["; inner ls; print_string "]"

let print_nodes nodes = print_list (print_int) @@ Array.to_list nodes

let bfs n graph nodes =
  let q = Queue.create () in
  Queue.push n q;
  let rec inner () =
    if Queue.is_empty q then (-1)
    else
      begin
        let v = Queue.pop q in
        if v = 1 then get nodes v
        else 
          begin
            List.iter (fun l -> 
                if get nodes l = (-1) || get nodes l > (get nodes v) + 1 then
                  (Queue.push l q; set nodes l ((get nodes v) + 1)))
              (get graph v);
(*            print_nodes nodes; print_string "\n";*)
            inner ()
          end
      end
  in inner ()

let solve n =
  let nodes = Array.make n (-1) in
  set nodes n 1;
  let g = gen_graph n in
  print_int @@ bfs n g nodes

let _ =
  let n = int_of_string @@ read_line() in
  solve n
0