JohnCosta27/AdventOfCode
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf(day7): small but easy pruning

Browse Source
This commit is contained in:
John Costa
2024-12-07 11:20:56 +00:00
parent 780816fda9
commit ee88542095
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
AdventOfCode2024/lib/day_07.ml
View File

@ -37,15 +37,16 @@ let attempt_solve_part2 { equals; numbers } =
match numbers with match numbers with
| head :: [] -> [ head ] | head :: [] -> [ head ]
| head :: tail -> | head :: tail ->
let plus = List.map (fun i -> i + head) (loop tail) in let below_solution = List.filter (fun i -> i <= equals) (loop tail) in
let times = List.map (fun i -> i * head) (loop tail) in let plus = List.map (fun i -> i + head) below_solution in
let times = List.map (fun i -> i * head) below_solution in
let concat = let concat =
List.map List.map
(fun i -> (fun i ->
let i_string = string_of_int i in let i_string = string_of_int i in
let head_string = string_of_int head in let head_string = string_of_int head in
int_of_string (i_string ^ head_string)) int_of_string (i_string ^ head_string))
(loop tail) below_solution
in in
plus @ times @ concat plus @ times @ concat
| _ -> raise (Invalid_argument "") | _ -> raise (Invalid_argument "")