aboutsummaryrefslogtreecommitdiff
path: root/2021/day-07/day-07.rkt
diff options
context:
space:
mode:
Diffstat (limited to '2021/day-07/day-07.rkt')
-rw-r--r--2021/day-07/day-07.rkt30
1 files changed, 30 insertions, 0 deletions
diff --git a/2021/day-07/day-07.rkt b/2021/day-07/day-07.rkt
new file mode 100644
index 0000000..81a2e4f
--- /dev/null
+++ b/2021/day-07/day-07.rkt
@@ -0,0 +1,30 @@
+#lang racket
+(require advent-of-code
+ threading
+ math/statistics)
+
+(define crab-data
+ (~> (open-aoc-input (find-session) 2021 7 #:cache #t)
+ port->string
+ string-trim
+ (string-split ",")
+ (map string->number _)))
+
+(define (gauss-sum n) (/ (* n (+ n 1)) 2))
+(define (compute-fuel-use f crabs align-to)
+ (for/sum ([crab (in-list crabs)])
+ (f (abs (- crab align-to)))))
+
+;; using the fact that the optimum location is at the median
+;; of the crabs' starting location for the linear relationship
+;; and at a coordinate within 1 unit of the mean for the quadratic one
+
+(~>> crab-data
+ (median <)
+ (compute-fuel-use identity crab-data))
+
+(~>> crab-data
+ mean
+ ((λ (m) (list (floor m) (ceiling m))))
+ (map (curry compute-fuel-use gauss-sum crab-data))
+ (apply min)) \ No newline at end of file