Add day2 in OCaml

2 files changed, 43 insertions, 0 deletions

diff --git a/2023/ocaml/day2/bin/dune b/2023/ocaml/day2/bin/dune
new file mode 100644
index 0000000..1fdd987
--- /dev/null
+++ b/2023/ocaml/day2/bin/dune
@@ -0,0 +1,4 @@
+(executable
+ (public_name day2)
+ (name main)
+ (libraries day2))
diff --git a/2023/ocaml/day2/bin/main.ml b/2023/ocaml/day2/bin/main.ml
new file mode 100644
index 0000000..6dba854
--- /dev/null
+++ b/2023/ocaml/day2/bin/main.ml
@@ -0,0 +1,39 @@
+let input_file = "input.txt"
+
+let read_file f =
+    let ic = open_in f in
+    try
+        let lines = really_input_string ic (in_channel_length ic) in
+        let lines = String.trim lines in
+        close_in ic ;
+        lines
+    with e ->
+        close_in_noerr ic;
+        raise e
+
+let rec parse ?(result=0) body =
+    match body with
+    | [] -> result
+    | x :: tail -> 
+        let plays = String.split_on_char ':' x
+        |> List.rev
+        |> List.hd
+        |> String.split_on_char ';'
+        |> List.map (String.split_on_char ',')
+        |> List.map (List.map @@ String.trim)
+        |> List.map (List.map (fun x -> (List.hd (String.split_on_char ' ' x) |> int_of_string, String.get (List.nth (String.split_on_char ' ' x) 1 ) 0)))
+        in 
+
+
+        let points = List.map (List.map (fun (p,c) -> if (c = 'r' && p > 12) || (c = 'g' && p > 13) || (c = 'b' && p > 14) then 0 else 1)) plays in
+        let add = if List.length (List.filter (fun x -> List.length x > 0) @@ List.map (List.filter (fun x -> x = 0)) points) = 0 then
+            String.split_on_char ':' x |> List.hd |> String.split_on_char ' ' |> List.rev |> List.hd |> int_of_string
+        else
+            0 in
+
+        parse tail ~result:(result+add)
+
+
+let () = 
+    let result = parse (String.split_on_char '\n' (read_file input_file)) in
+    Printf.printf "%d" result