blob: 4c6334d1314577f7fa9a8caaa4ec941ab5af29ed (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
#lang racket
(require "../../jj-aoc.rkt"
threading
racket/struct)
(struct coord (x y z) #:transparent)
(define (coord-broadcast f c1 c2)
(match-define (coord x1 y1 z1) c1)
(match-define (coord x2 y2 z2) c2)
(coord (f x1 x2) (f y1 y2) (f z1 z2)))
(define (coord-reduce f c1 c2)
(foldl (λ (i1 i2 acc) (+ acc (f i1 i2))) 0 (struct->list c1) (struct->list c2)))
(define coord-delta (curry coord-broadcast -))
(define coord-sum (curry coord-broadcast +))
(define coord-manhattan (curry coord-reduce (coord 0 0 0)))
(define (create-scan-data d)
(for/list ([l (in-list d)])
(for/list ([pt (in-list (~> (string-split l "\n") rest))])
(~>> pt
(regexp-match #px"(-?\\d+),(-?\\d+),(-?\\d+)")
rest
(map string->number)
(apply coord)))))
(define scanner-data (create-scan-data (~>> (open-day 19 2021) port->string (string-split _ "\n\n"))))
(define (generate-rotations scanner)
(apply
map
list
(for*/list ([pt (in-list scanner)])
(match-define (coord x y z) pt)
(define orientations (list (list x y z) (list x z (- y)) (list x (- y) (- z)) (list x (- z) y)))
(append* (for/list ([o (in-list orientations)])
(match-define (list x* y* z*) o)
(list (list x y z)
(list (- x) z y)
(list z (- y) x)
(list y x (- z))
(list (- y) (- z) x)))))))
(define (find-overlaps scan1 scan2)
(for/list ([rotation (in-permutations scan2)])
(map coord-sum scan1 rotation)))
|
