summaryrefslogtreecommitdiff
path: root/ocaml/lib/application/d01p02.ml
diff options
context:
space:
mode:
Diffstat (limited to 'ocaml/lib/application/d01p02.ml')
-rw-r--r--ocaml/lib/application/d01p02.ml55
1 files changed, 55 insertions, 0 deletions
diff --git a/ocaml/lib/application/d01p02.ml b/ocaml/lib/application/d01p02.ml
new file mode 100644
index 0000000..7910486
--- /dev/null
+++ b/ocaml/lib/application/d01p02.ml
@@ -0,0 +1,55 @@
+open Domain.D01p02
+
+let err_empty_input = "input is empty"
+
+let rec input_to_lines input =
+ let lines =
+ String.split_on_char '\n' input |> List.filter (fun l -> l <> "")
+ in
+ pp_lines lines ; lines
+
+and pp_lines = function
+ | [] ->
+ ()
+ | h :: t ->
+ Logs.debug (fun m -> m "Line=%s" h) ;
+ pp_lines t
+
+let parse_line line =
+ Scanf.sscanf line "%c%d" (fun d n ->
+ match d with
+ | 'L' ->
+ Dial.Left n
+ | 'R' ->
+ Dial.Right n
+ | _ ->
+ failwith "invalid direction" )
+
+let lines_to_turns =
+ List.map (fun (l : string) ->
+ Logs.debug (fun m -> m "parsing %s into turn" l) ;
+ let turn = parse_line l in
+ Logs.debug (fun m -> m "%s" (Dial.pp_direction turn)) ;
+ turn )
+
+let rec turn_aux count dial = function
+ | [] ->
+ count
+ | h :: t -> (
+ let dial_and_count = Turner.turn dial count h in
+ match dial_and_count with
+ | dial', count' ->
+ Logs.debug (fun m ->
+ m "(dial=%d, count=%d)" (Dial.to_int dial') count' ) ;
+ turn_aux count' dial' t )
+
+let turn current = turn_aux 0 current
+
+let solve input () =
+ if input = "" then Error err_empty_input
+ else
+ let lines = input_to_lines input in
+ let turns = lines_to_turns lines in
+ let start = Dial.start in
+ let answer = turn start turns in
+ Ok answer