diff options
| -rw-r--r-- | AoC2023/day14/solve.lisp | 45 |
1 files changed, 20 insertions, 25 deletions
diff --git a/AoC2023/day14/solve.lisp b/AoC2023/day14/solve.lisp index 9f9d6e8..e476bd5 100644 --- a/AoC2023/day14/solve.lisp +++ b/AoC2023/day14/solve.lisp @@ -16,22 +16,17 @@ (< -1 col maxcol)))) (defun roll-rocks (field direction) - (loop for row below (array-dimension field 0) - sum - (loop for col below (array-dimension field 1) - for (new-row new-col) = (roll direction row col) - when (and - (in-bounds field new-row new-col) - (eq #\O (aref field row col)) - (eq #\. (aref field new-row new-col))) - do (setf (aref field new-row new-col) #\O - (aref field row col) #\.) - and count t))) - -(defun roll-rocks-full (field direction) - (loop - for result = (roll-rocks field direction) - until (zerop result))) + (loop for i below (array-dimension field 0) do + (loop for j below (array-dimension field 1) do + (loop + for (row col) = (list i j) then (list new-row new-col) + for (new-row new-col) = (roll direction row col) + while (and + (in-bounds field new-row new-col) + (eq #\O (aref field row col)) + (eq #\. (aref field new-row new-col))) + do (setf (aref field new-row new-col) #\O + (aref field row col) #\.))))) (defun array-slice (arr row) (make-array (array-dimension arr 1) @@ -48,17 +43,16 @@ (let* ((rows (length input)) (field (make-array (list rows (length (car input))) :initial-contents input))) - (roll-rocks-full field 'north) + (roll-rocks field 'north) (north-load field))) -(defun draw-field (field) +(defun draw-field (field &optional (out *standard-output*)) (destructuring-bind (maxrow maxcol) (array-dimensions field) - (with-output-to-string (out) - - (loop for i below maxrow do - (loop for j below maxcol do - (write-char (aref field i j) out)) - (terpri out))))) + (loop for i below maxrow do + (loop for j below maxcol do + (write-char (aref field i j) out)) + (terpri out)) + (terpri out))) (defun solve2 (input iters) (let* ((rows (length input)) @@ -76,6 +70,7 @@ ;; (solve2 (uiop:read-file-lines "input") 100) (fiveam:test solutions (fiveam:is (= 136 (solve1 (uiop:read-file-lines "eg-in")))) - (fiveam:is (= 108935 (solve1 (uiop:read-file-lines "input"))))) + (fiveam:is (= 108935 (solve1 (uiop:read-file-lines "input")))) + ) (fiveam:run!) |