blob: d4e3185f4c45480638070db46d11498b5497d5b5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
#lang racket
(require advent-of-code)
(define raw-packets
(parameterize ([current-readtable (make-readtable #f #\, #\space #f)])
(port->list read (open-aoc-input (find-session) 2022 13 #:cache #true))))
(define (compare xs ys)
(match* (xs ys)
[('() (list* _)) #true]
[((list* _) '()) #false]
[((list* a x-rest) (list* a y-rest)) (compare x-rest y-rest)]
[((list* (? integer? x) _) (list* (? integer? y) _)) (< x y)]
[((list* (? list? xs*) _) (list* (? list? ys*) _)) (compare xs* ys*)]
[((list* (? list?) _) (list* (? integer? y) y-rest)) (compare xs (cons (list y) y-rest))]
[((list* (? integer? x) x-rest) (list* (? list?) _)) (compare (cons (list x) x-rest) ys)]))
;; part 1
(for/sum ([i (in-naturals 1)] [packet (in-slice 2 raw-packets)] #:when (apply compare packet)) i)
;; part 2
(define divider-packets (list '((2)) '((6))))
(define amended-packets (append divider-packets raw-packets))
(for/product ([i (in-naturals 1)] [packet (in-list (sort amended-packets compare))]
#:when (member packet divider-packets))
i)
|
