Merge branch 'main' of https://github.com/hunkyjimpjorps/AdventOfCode

1 files changed, 32 insertions, 0 deletions

diff --git a/aoc2023-racket/day-09/day-09.rkt b/aoc2023-racket/day-09/day-09.rkt
new file mode 100644
index 0000000..5eda1eb
--- /dev/null
+++ b/aoc2023-racket/day-09/day-09.rkt
@@ -0,0 +1,32 @@
+#lang racket
+
+(require advent-of-code
+         threading)
+
+(define histories
+  (for/list ([raw-history (in-lines (open-aoc-input (find-session) 2023 9 #:cache #true))])
+    (~>> raw-history 
+         string-split 
+         (map string->number))))
+
+(define (constant? xs) 
+  (= 1 (length (remove-duplicates xs))))
+
+(define/match (derivative xs)
+  [((list a b)) (list (- b a))]
+  [((list* a b _)) (cons (- b a) (derivative (rest xs)))])
+
+(define (extrapolate xs) 
+  (if (constant? xs) 
+      (car xs) 
+      (+ (last xs) (extrapolate (derivative xs)))))
+
+;; part 1
+(~>> histories 
+     (map extrapolate) 
+     (apply +))
+
+;; part 2
+(~>> histories 
+     (map (λ~> reverse extrapolate)) 
+     (apply +))