aboutsummaryrefslogtreecommitdiff
path: root/racket/aoc2021/day-10/day-10.rkt
diff options
context:
space:
mode:
Diffstat (limited to 'racket/aoc2021/day-10/day-10.rkt')
-rw-r--r--racket/aoc2021/day-10/day-10.rkt57
1 files changed, 57 insertions, 0 deletions
diff --git a/racket/aoc2021/day-10/day-10.rkt b/racket/aoc2021/day-10/day-10.rkt
new file mode 100644
index 0000000..ea1b389
--- /dev/null
+++ b/racket/aoc2021/day-10/day-10.rkt
@@ -0,0 +1,57 @@
+#lang racket
+
+(require math/statistics
+ threading
+ "../../jj-aoc.rkt")
+
+(define chunks (port->lines (open-day 10 2021)))
+
+(define (opening-bracket? c)
+ (member c (string->list "([{<")))
+
+(define (matching-brackets? c-left c-right)
+ (member (string c-left c-right) '("()" "[]" "{}" "<>")))
+
+(define (parse-brackets lst [acc '()])
+ (cond
+ [(empty? lst) acc]
+ [(opening-bracket? (first lst)) (parse-brackets (rest lst) (cons (first lst) acc))]
+ [(matching-brackets? (first acc) (first lst)) (parse-brackets (rest lst) (rest acc))]
+ [else (get-score (first lst))]))
+
+;; part 1
+(define (get-score c)
+ (match (string c)
+ [")" 3]
+ ["]" 57]
+ ["}" 1197]
+ [">" 25137]))
+
+(define (score-invalid-string chunk)
+ (match (parse-brackets (string->list chunk))
+ [(? list?) 0]
+ [n n]))
+
+(for/sum ([chunk (in-list chunks)]) (score-invalid-string chunk))
+
+;; part 2
+(define (completion-score lst)
+ (for/fold ([score 0]) ([c (in-list lst)])
+ (define val
+ (match (string c)
+ ["(" 1]
+ ["[" 2]
+ ["{" 3]
+ ["<" 4]))
+ (+ (* 5 score) val)))
+
+(define (score-incomplete-string chunk)
+ (match (parse-brackets (string->list chunk))
+ [(? list? lst) (completion-score lst)]
+ [n 0]))
+
+(~>> (for*/list ([chunk (in-list chunks)]
+ [score (in-value (score-incomplete-string chunk))]
+ #:when (> score 0))
+ score)
+ (median <))