From a4cf01cb5152ebb738b466b3e8a26b7ed98421fb Mon Sep 17 00:00:00 2001 From: drew Date: Fri, 8 Dec 2023 19:07:16 +0800 Subject: [PATCH] Use regex --- lib/year2023/day08.ml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/year2023/day08.ml b/lib/year2023/day08.ml index be40179..61bbe6e 100644 --- a/lib/year2023/day08.ml +++ b/lib/year2023/day08.ml @@ -60,8 +60,8 @@ let solve { dirs; map } = ;; let solve' { dirs; map } = - let is_start s = s |> String.to_list_rev |> List.hd_exn |> Char.equal 'A' in - let is_end s = s |> String.to_list_rev |> List.hd_exn |> Char.equal 'Z' in + let is_start s = Pcre.pmatch ~rex:(Pcre.regexp {|A\z|}) s in + let is_end s = Pcre.pmatch ~rex:(Pcre.regexp {|Z\z|}) s in let starts = map |> Map.keys |> List.filter ~f:is_start in let make_move (count, steps, nodes) dir = let ends, rest = List.partition_tf nodes ~f:is_end in