diff options
Diffstat (limited to 'racket/aoc2021/day-15/day-15-list-nodes.rkt')
-rw-r--r-- | racket/aoc2021/day-15/day-15-list-nodes.rkt | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/racket/aoc2021/day-15/day-15-list-nodes.rkt b/racket/aoc2021/day-15/day-15-list-nodes.rkt new file mode 100644 index 0000000..38c558a --- /dev/null +++ b/racket/aoc2021/day-15/day-15-list-nodes.rkt @@ -0,0 +1,67 @@ +#lang racket +(require "../../jj-aoc.rkt" + threading + graph) + +(define data + (for/fold ([cells (hash)]) + ([row (in-lines (open-day 15 2021))] + [x (in-naturals)]) + (for/fold ([cells cells]) + ([n (in-string row)] + [y (in-naturals)]) + (hash-set cells + `(,x ,y) + (~> n string string->number))))) + +(define x-max (~>> data hash-keys (map first) (apply max))) +(define y-max (~>> data hash-keys (map second) (apply max))) + +(define (neighbors pt d) + (match-define (list x y) pt) + (~>> (list (list (add1 x) y) + (list (sub1 x) y) + (list x (add1 y)) + (list x (sub1 y))) + (filter (curry hash-has-key? d)))) + +(define (grid-graph d) + (weighted-graph/directed + (for*/list ([coord (in-list (hash-keys d))] + [neighbor (in-list (neighbors coord d))] + [weight (in-value (hash-ref d neighbor))]) + (list weight coord neighbor)))) + +;; part 1 +(define (find-path-weight d) + (define grid (grid-graph d)) + (let-values ([(node-distances _) (dijkstra grid '(0 0))]) + (define xm (~>> d hash-keys (map first) (apply max))) + (define ym (~>> d hash-keys (map second) (apply max))) + (hash-ref node-distances (list xm ym)))) + +(~> data + find-path-weight + time) + +;; part 2 +(define nine-cycle + (in-cycle (inclusive-range 1 9))) + +(define (expand-data data) + (for/fold ([cells (hash)]) + ([coord (in-list (hash-keys data))]) + (match-define (list x y) coord) + (for*/fold ([cells cells]) + ([n (in-range 5)] + [m (in-range 5)] + [val (in-value (hash-ref data coord))]) + (hash-set cells + (list (+ x (* n (add1 x-max))) + (+ y (* m (add1 y-max)))) + (sequence-ref nine-cycle (+ val n m -1)))))) + +(~> data + expand-data + find-path-weight + time)
\ No newline at end of file |