aboutsummaryrefslogtreecommitdiff
path: root/racket/aoc2021/day-02/day-02.rkt
diff options
context:
space:
mode:
Diffstat (limited to 'racket/aoc2021/day-02/day-02.rkt')
-rw-r--r--racket/aoc2021/day-02/day-02.rkt24
1 files changed, 24 insertions, 0 deletions
diff --git a/racket/aoc2021/day-02/day-02.rkt b/racket/aoc2021/day-02/day-02.rkt
new file mode 100644
index 0000000..0bd0c3d
--- /dev/null
+++ b/racket/aoc2021/day-02/day-02.rkt
@@ -0,0 +1,24 @@
+#lang racket
+(require advent-of-code
+ threading
+ algorithms)
+
+(define motion-data
+ (~> (open-aoc-input (find-session) 2021 2 #:cache (string->path "./cache"))
+ (port->list read _)
+ (chunks-of _ 2)))
+
+;; part 1
+(for/fold ([depth 0] [position 0] #:result (* depth position)) ([motion (in-list motion-data)])
+ (match motion
+ [(list 'forward x) (values depth (+ position x))]
+ [(list 'up x) (values (- depth x) position)]
+ [(list 'down x) (values (+ depth x) position)]))
+
+;; part 2
+(for/fold ([aim 0] [depth 0] [position 0] #:result (* depth position))
+ ([motion (in-list motion-data)])
+ (match motion
+ [(list 'forward x) (values aim (+ depth (* aim x)) (+ position x))]
+ [(list 'up x) (values (- aim x) depth position)]
+ [(list 'down x) (values (+ aim x) depth position)]))