aboutsummaryrefslogtreecommitdiff
path: root/racket/aoc2021/day-06/day-06.rkt
diff options
context:
space:
mode:
Diffstat (limited to 'racket/aoc2021/day-06/day-06.rkt')
-rw-r--r--racket/aoc2021/day-06/day-06.rkt27
1 files changed, 27 insertions, 0 deletions
diff --git a/racket/aoc2021/day-06/day-06.rkt b/racket/aoc2021/day-06/day-06.rkt
new file mode 100644
index 0000000..d8855ba
--- /dev/null
+++ b/racket/aoc2021/day-06/day-06.rkt
@@ -0,0 +1,27 @@
+#lang racket
+(require advent-of-code
+ list-utils
+ threading
+ racket/hash)
+
+(define fish-data
+ (~> (open-aoc-input (find-session) 2021 6 #:cache (string->path "./cache"))
+ port->string
+ string-trim
+ (string-split ",")
+ (map string->number _)))
+
+(define (simulate-fish time-period)
+ (for/fold ([state (frequencies fish-data)] #:result (~> state hash-values (apply + _)))
+ ([day (inclusive-range 1 time-period)])
+ (define day-older-fish
+ (for/hash ([(days pop) (in-hash state)])
+ (values (sub1 days) pop)))
+ (define breeding-fish (hash-ref day-older-fish -1 0))
+ (hash-union (hash-remove day-older-fish -1) (hash 8 breeding-fish 6 breeding-fish) #:combine +)))
+
+;; part 1
+(simulate-fish 80)
+
+;; part 2
+(simulate-fish 256)