adventofcode
/
2021
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Day 05: Now purely functional!

This commit is contained in:
Jonathan Chan 2021-12-04 22:43:10 -08:00
parent bb5718c3ec
commit 8d88893d2a
1 changed files with 27 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/05.rkt
View File

@ -15,30 +15,35 @@
(define (count-crosses grid) (define (count-crosses grid)
(count #{> % 1} (hash-values grid))) (count #{> % 1} (hash-values grid)))
(define grid (make-hash)) (define grid1
(for/fold ([grid (hash)])
([hv hvs])
(match hv
[`(,x ,y1 ,x ,y2)
(for/fold ([grid grid])
([y (range (min y1 y2) (add1 (max y1 y2)))])
(hash-update grid `(,x ,y) add1 0))]
[`(,x1 ,y ,x2 ,y)
(for/fold ([grid grid])
([x (range (min x1 x2) (add1 (max x1 x2)))])
(hash-update grid `(,x ,y) add1 0))])))
(for ([hv hvs]) (define part1 (count-crosses grid1))
(match hv
[`(,x ,y1 ,x ,y2)
(for ([y (range (min y1 y2) (add1 (max y1 y2)))])
(hash-update! grid `(,x ,y) add1 0))]
[`(,x1 ,y ,x2 ,y)
(for ([x (range (min x1 x2) (add1 (max x1 x2)))])
(hash-update! grid `(,x ,y) add1 0))]))
(define part1 (count-crosses grid)) (define grid2
(for/fold ([grid grid1])
([diag diags])
(match diag
[`(,x1 ,y1 ,x2 ,y2)
(define-values (x-offset x-step)
(if (>= x2 x1) (values add1 1) (values sub1 -1)))
(define-values (y-offset y-step)
(if (>= y2 y1) (values add1 1) (values sub1 -1)))
(for/fold ([grid grid])
([x (range x1 (x-offset x2) x-step)]
[y (range y1 (y-offset y2) y-step)])
(hash-update grid `(,x ,y) add1 0))])))
(for ([diag diags]) (define part2 (count-crosses grid2))
(match diag
[`(,x1 ,y1 ,x2 ,y2)
(define-values (x-offset x-step)
(if (>= x2 x1) (values add1 1) (values sub1 -1)))
(define-values (y-offset y-step)
(if (>= y2 y1) (values add1 1) (values sub1 -1)))
(for ([x (range x1 (x-offset x2) x-step)]
[y (range y1 (y-offset y2) y-step)])
(hash-update! grid `(,x ,y) add1 0))]))
(define part2 (count-crosses grid))
(show-solution part1 part2) (show-solution part1 part2)