aboutsummaryrefslogtreecommitdiff
path: root/racket/aoc2020/day-10/day-10.rkt
diff options
context:
space:
mode:
Diffstat (limited to 'racket/aoc2020/day-10/day-10.rkt')
-rw-r--r--racket/aoc2020/day-10/day-10.rkt37
1 files changed, 37 insertions, 0 deletions
diff --git a/racket/aoc2020/day-10/day-10.rkt b/racket/aoc2020/day-10/day-10.rkt
new file mode 100644
index 0000000..77d9bb7
--- /dev/null
+++ b/racket/aoc2020/day-10/day-10.rkt
@@ -0,0 +1,37 @@
+#lang racket
+
+(require advent-of-code
+ threading
+ algorithms
+ memoize)
+
+;; part 1
+(define adapters
+ (~> (fetch-aoc-input (find-session) 2020 10)
+ (string-split "\n")
+ (map string->number _)
+ (sort <)
+ ((λ (xs) (flatten (list 0 xs (+ 3 (last xs))))))))
+
+(~>> adapters
+ (sliding _ 2)
+ (map (match-lambda
+ [(list b a) (- a b)]))
+ (group-by identity)
+ (map length)
+ (apply *))
+
+;; part 2
+(define subpaths
+ (for*/hash ([adapter (in-list adapters)])
+ (define predecessor-candidates (inclusive-range (+ 1 adapter) (+ 3 adapter)))
+ (values adapter (filter (λ (p) (member p adapters)) predecessor-candidates))))
+
+(define/memo (find-paths from to)
+ (define paths (hash-ref subpaths from 'failed))
+ (match paths
+ ['failed 0]
+ [(list-no-order (== to) _ ...) 1]
+ [ts (for/sum ([t (in-list ts)]) (find-paths t to))]))
+
+(find-paths (first adapters) (last adapters))