···
read [] |> List.rev |> String.concat ""
14
-
let regex = Str.regexp "mul(\\([0-9]+\\),\\([0-9]+\\))"
14
+
let regex_mul = Str.regexp "mul(\\([0-9]+\\),\\([0-9]+\\))"
let regex_mul_part_2 = Str.regexp "mul(\\([0-9]+\\),\\([0-9]+\\))\\|don't()"
let regex_do_part_2 = Str.regexp "do()"
18
-
let calculate_total str =
19
-
let rec calculate str acc =
21
-
let _ = Str.search_forward regex str 0 in
22
-
let a = Str.matched_group 1 str |> int_of_string in
23
-
let b = Str.matched_group 2 str |> int_of_string in
25
-
let result = a * b in
26
-
let new_str = Str.replace_first regex (string_of_int result) str in
27
-
calculate new_str (acc + result)
28
-
with Not_found -> acc
32
-
let calculate_total_part_2 str =
18
+
let calculate_total str regexp =
let rec calculate str regexp acc =
let _ = Str.search_forward regexp str 0 in
···
| cmd when String.starts_with ~prefix:"mul" cmd ->
let a = Str.matched_group 1 str |> int_of_string in
let b = Str.matched_group 2 str |> int_of_string in
46
-
let result = a * b in
47
-
calculate new_str regexp (acc + result)
32
+
calculate new_str regexp (acc + (a * b))
51
-
calculate str regex_mul_part_2 0
36
+
calculate str regexp 0
let ic = open_in filename in
55
-
read_file ic |> calculate_total |> Printf.sprintf "Sum %d" |> print_endline;
40
+
let data = read_file ic in
41
+
calculate_total data regex_mul |> Printf.sprintf "Sum %d" |> print_endline;
57
-
let ic = open_in filename in
58
-
read_file ic |> calculate_total_part_2
43
+
calculate_total data regex_mul_part_2
|> Printf.sprintf "Part 2 Sum %d"