src/2018/days/day-08.lisp @ 8b272a80df17
Ugh
author |
Steve Losh <steve@stevelosh.com> |
date |
Thu, 12 Dec 2019 21:08:08 -0500 |
parents |
cd781337a694 |
children |
182bdd87fd9e |
(defpackage :advent/2018/08 #.cl-user::*advent-use*)
(in-package :advent/2018/08)
(named-readtables:in-readtable :interpol-syntax)
(defstruct (node (:conc-name nil))
children metadata)
(defun read-node (stream)
(let ((children-count (read stream))
(metadata-count (read stream)))
(make-node :children (iterate
(repeat children-count)
(collect (read-node stream) :result-type vector))
:metadata (iterate
(repeat metadata-count)
(collect (read stream))))))
(defun node-value (node &aux (children (children node)))
(if (emptyp children)
(summation (metadata node))
(iterate
(for meta :in (metadata node))
(for index = (1- meta))
(when (array-in-bounds-p children index)
(summing (node-value (aref children index)))))))
(define-problem (2018 8) (data) (37905 33891)
(let ((root (read-node data)))
(values
(recursively ((node root))
(+ (summation (metadata node))
(summation (children node) :key #'recur)))
(node-value root))))