結果

問題 No.3 ビットすごろく
ユーザー kashiwagi513
提出日時 2019-03-01 11:44:27
言語 OCaml
(5.2.1)
結果
WA  
実行時間 -
コード長 1,052 bytes
コンパイル時間 359 ms
コンパイル使用メモリ 19,916 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-09 00:34:23
合計ジャッジ時間 1,333 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 14 WA * 19
権限があれば一括ダウンロードができます

ソースコード

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 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) then Queue.push l q; set nodes l ((get nodes v) + 1))
              (get graph v);
            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