diff options
-rw-r--r-- | aoc2023-other/day-03/day-03.rkt | 68 | ||||
-rw-r--r-- | aoc2023/src/day3/solve.gleam | 13 |
2 files changed, 72 insertions, 9 deletions
diff --git a/aoc2023-other/day-03/day-03.rkt b/aoc2023-other/day-03/day-03.rkt new file mode 100644 index 0000000..d52b11b --- /dev/null +++ b/aoc2023-other/day-03/day-03.rkt @@ -0,0 +1,68 @@ +#lang racket + +(require advent-of-code + threading) + +(struct posn (x y) #:transparent) +(struct part (n posns) #:transparent) + +(define (make-board port) + (for*/hash ([(row y) (in-indexed (port->lines port))] + [(col x) (in-indexed (string->list row))] + #:unless (equal? col #\.)) + (define v + (cond + [(string->number (string col))] + [(equal? col #\*) 'gear] + [else 'other])) + (values (posn x y) v))) + +(define board (~> (open-aoc-input (find-session) 2023 3 #:cache #true) make-board)) + +(define (posn<? a b) + (match-define (list (cons (posn a-x a-y) _) (cons (posn b-x b-y) _)) (list a b)) + (if (= a-y b-y) (< a-x b-x) (< a-y b-y))) + +(define (find-cells f b) + (~> (for/hash ([(k v) (in-hash b)] #:when (f v)) + (values k v)) + hash->list + (sort posn<?))) + +(define (group-into-parts cells [acc '()]) + (match* (cells acc) + [('() acc) acc] + [((list* (cons (and pt (posn x y)) n) cs) (list* (part n* (and pts (list* (posn x* y) rest-pts))) + rest-acc)) + #:when (= (- x x*) 1) + (group-into-parts cs (cons (part (+ n (* n* 10)) (cons pt pts)) rest-acc))] + [((list* (cons pt n) cs) acc) (group-into-parts cs (cons (part n (list pt)) acc))])) + +(define (neighbors p) + (for*/list ([dx '(-1 0 1)] [dy '(-1 0 1)] #:unless (and (= dx 0) (= dy 0))) + (posn (+ dx (posn-x p)) (+ dy (posn-y p))))) + +(define to-neighbors (λ~>> part-posns (append-map neighbors) remove-duplicates)) +(define (symbol-in-neighbors b pt acc) + (~>> pt + to-neighbors + (ormap (λ (p) + (let ([lookup (hash-ref b p #f)]) (or (equal? lookup 'gear) (equal? lookup 'other))))) + ((λ (bool) (if bool (+ acc (part-n pt)) acc))))) + +;; part 1 +(define parts (~>> board (find-cells integer?) group-into-parts)) +(foldl (curry symbol-in-neighbors board) 0 parts) + +;; part 2 +(define gears (~>> board (find-cells (curry equal? 'gear)) (map car))) +(define parts-with-neighbors + (~>> parts (map (λ (pt) (struct-copy part pt [posns (to-neighbors pt)]))))) + +(define (find-parts-near-gear pts gear) + (filter-map (λ (pt) (and (findf (curry equal? gear) (part-posns pt)) (part-n pt))) pts)) + +(~>> gears + (filter-map (λ~>> (find-parts-near-gear parts-with-neighbors) + ((λ (ns) (if (= (length ns) 2) (* (first ns) (second ns)) #f))))) + (apply +)) diff --git a/aoc2023/src/day3/solve.gleam b/aoc2023/src/day3/solve.gleam index 21acd26..fa16bb8 100644 --- a/aoc2023/src/day3/solve.gleam +++ b/aoc2023/src/day3/solve.gleam @@ -5,7 +5,6 @@ import gleam/string import gleam/list import gleam/int import gleam/order.{type Order, Eq} -import gleam/result type Coord { Coord(x: Int, y: Int) @@ -102,7 +101,7 @@ fn all_neighbors(c: Coord) -> List(Coord) { } } -fn check_part_neighbors(part: Part, board: Board) -> Result(Int, Nil) { +fn sum_valid_parts(acc: Int, part: Part, board: Board) -> Int { let neighbors = part.coords |> list.flat_map(all_neighbors) @@ -110,8 +109,8 @@ fn check_part_neighbors(part: Part, board: Board) -> Result(Int, Nil) { let sym = [Ok(Symbol(Gear)), Ok(Symbol(SomethingElse))] case list.any(neighbors, fn(c) { list.contains(sym, dict.get(board, c)) }) { - True -> Ok(part.part_number) - False -> Error(Nil) + True -> acc + part.part_number + False -> acc } } @@ -121,11 +120,7 @@ pub fn part1(input: String) -> Int { board |> find_all_part_digits |> to_parts - |> io.debug - |> list.map(check_part_neighbors(_, board)) - |> io.debug - |> result.values - |> int.sum + |> list.fold(0, fn(acc, p) { sum_valid_parts(acc, p, board) }) } fn to_part_with_neighbors(part: Part) -> Part { |