Use find_map

drewolson · Dec 13, 2023 · 76fa1ae · 76fa1ae
1 parent 7fb13c2
commit 76fa1ae
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/year2023/day13.ml b/lib/year2023/day13.ml
@@ -56,9 +56,8 @@ let reflection_score' g =
     with_swap coord ~f:(fun _ ->
       g
       |> reflection_score
-      |> List.filter_map ~f:(fun (axis, score) ->
-        if equal_axis axis old_axis then None else Some score)
-      |> List.hd))
+      |> List.find_map ~f:(fun (axis, score) ->
+        if equal_axis axis old_axis then None else Some score)))
   |> Option.value_exn
 ;;