diff options
Diffstat (limited to '2021/day-07/day-07.rkt')
-rw-r--r-- | 2021/day-07/day-07.rkt | 30 |
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 |