blob: 32d99776f486d7ff5079bcbf43602cee26696cbf (
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
(ql:quickload '(fiveam uiop))
(defun coord-x-minor (width height)
(lambda (x y)
(when (and (< -1 x width)
(< -1 y height))
(+ x (* y width)))))
(defun forest (data)
(let ((width (length (car data)))
(height (length data)))
(values width height
(map 'vector
(lambda (x) (- (char-code x) 48))
(apply #'concatenate 'string data)))))
(defun solver-p1 (filename)
(multiple-value-bind (width height forest-arr) (forest (uiop:read-file-lines filename))
(let ((visibility-mask (make-array (* width height) :initial-element nil))
(location-x (coord-x-minor width height)))
(flet ((toggle-visibility (major minor location reverse)
(let ((range (if reverse
(loop for i downfrom (1- minor) to 0 collect i)
(loop for i below minor collect i))))
(dotimes (m major)
(loop
for n in range
for maxh = -1 then (max maxh tree-height)
for tree-height = (aref forest-arr (funcall location n m))
when (> tree-height maxh)
do (setf (aref visibility-mask (funcall location n m)) t)))))
(location-y (y x) (funcall location-x x y)))
(toggle-visibility height width location-x nil)
(toggle-visibility height width location-x t)
(toggle-visibility width height #'location-y nil)
(toggle-visibility width height #'location-y t))
(loop for v across visibility-mask counting v))))
(defun solver-p2 (filename)
(multiple-value-bind (width height forest-arr) (forest (uiop:read-file-lines filename))
(let ((score (make-array (* width height)))
(location-x (coord-x-minor width height)))
(flet ((score-direction (base dir)
(loop for l in dir
for tree-height = (aref forest-arr l)
count l until (>= tree-height base))))
(dotimes (y height)
(dotimes (x width)
(let ((base (aref forest-arr (funcall location-x x y))))
(setf (aref score (funcall location-x x y))
(*
(score-direction base (loop for l from (1+ x) below width collect (funcall location-x l y))) ;; to right
(score-direction base (loop for l from (1+ y) below height collect (funcall location-x x l))) ;; to bottom
(score-direction base (loop for l downfrom (1- x) to 0 collect (funcall location-x l y))) ;; to left
(score-direction base (loop for l downfrom (1- y) to 0 collect (funcall location-x x l)))))))) ;; to top
(loop for v across score maximize v)))))
(fiveam:test solutions
(fiveam:is (= 21 (solver-p1 "eg-in")))
(fiveam:is (= 1672 (solver-p1 "input")))
(fiveam:is (= 8 (solver-p2 "eg-in")))
(fiveam:is (= 327180 (solver-p2 "input"))))
(fiveam:run-all-tests)
|
