haskal
/
aoc2020
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

day 12: use #:result

This commit is contained in:
xenia 2020-12-12 01:02:03 -05:00
parent f8f7363dba
commit a8825b52cb
1 changed files with 22 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
12.rkt
View File

@ -7,10 +7,11 @@
;; helper functions here
(define d->r degrees->radians)
(define (complex-manhat s)
(inexact->exact (round (+ (abs (real-part s)) (abs (imag-part s))))))
(define (part1 input)
(define-values [dir s]
(for/fold ([dir 0] [s 0]) ([ins (in-list input)])
(for/fold ([dir 0] [s 0] #:result (complex-manhat s)) ([ins (in-list input)])
(match-define (cons ltr num) ins)
(match ltr
["N" (values dir (+ s (* +i num)))]
@ -20,11 +21,9 @@
["L" (values (+ dir (d->r num)) s)]
["R" (values (- dir (d->r num)) s)]
["F" (values dir (+ s (make-polar num dir)))])))
(inexact->exact (round (+ (abs (real-part s)) (abs (imag-part s))))))
(define (part2 input)
(define-values [s w]
(for/fold ([s 0] [w 10+1i]) ([ins (in-list input)])
(for/fold ([s 0] [w 10+1i] #:result (complex-manhat s)) ([ins (in-list input)])
(match-define (cons ltr num) ins)
(match ltr
["N" (values s (+ w (* +i num)))]
@ -34,7 +33,6 @@
["L" (values s (* w (make-polar 1 (d->r num))))]
["R" (values s (* w (make-polar 1 (d->r (- num)))))]
["F" (values (+ s (* w num)) w)])))
(inexact->exact (round (+ (abs (real-part s)) (abs (imag-part s))))))
(module+ main
(define input