aoc2020/day1.ml

32 lines
843 B
OCaml
Raw Normal View History

2020-12-03 09:11:29 -05:00
module IntSet = Set_ext.Make(Int)
2020-12-02 04:31:32 -05:00
let read_ints =
Bracket.infile_lines ~line:int_of_string ~of_seq:IntSet.of_seq
2020-12-02 09:10:29 -05:00
let rec find ?(target=2020) ~n set =
if n < 1 then
raise (Invalid_argument "n must be at least 1")
else if n = 1 then
if IntSet.mem target set then Some [target] else None
else
let joined x =
Option.map (List.cons x) (find ~target:(target - x) ~n:(n - 1) set) in
IntSet.exists_opt joined set
let main' n input =
match find ~n (read_ints input) with
2020-12-03 09:11:29 -05:00
| Some xs -> Misc.print_prod xs
2020-12-02 09:10:29 -05:00
| None -> raise Not_found
2020-12-03 09:11:29 -05:00
let main = Misc.main 1 [|main' 2; main' 3|]
2020-12-02 09:10:29 -05:00
let%expect_test "part 1" =
2020-12-03 09:11:29 -05:00
main' 2 "../../data/day1"; (* is this always the right path??? *)
2020-12-02 09:10:29 -05:00
[%expect{| 534 * 1486 = 793524 |}]
let%expect_test "part 2" =
2020-12-03 09:11:29 -05:00
main' 3 "../../data/day1";
2020-12-02 09:10:29 -05:00
[%expect{| 71 * 686 * 1263 = 61515678 |}]