diff options
| -rw-r--r-- | AoC2022/07/solver.el | 58 |
1 files changed, 38 insertions, 20 deletions
diff --git a/AoC2022/07/solver.el b/AoC2022/07/solver.el index ab97c08..38354ff 100644 --- a/AoC2022/07/solver.el +++ b/AoC2022/07/solver.el @@ -6,10 +6,6 @@ ;; Maintainer: Óscar Nájera <hi@oscarnajera.com> ;; Created: December 08, 2022 ;; Modified: December 08, 2022 -;; Version: 0.0.1 -;; Keywords: abbrev bib c calendar comm convenience data docs emulations extensions faces files frames games hardware help hypermedia i18n internal languages lisp local maint mail matching mouse multimedia news outlines processes terminals tex tools unix vc wp -;; Homepage: https://github.com/titan/solver -;; Package-Requires: ((emacs "24.3")) ;; ;; This file is not part of GNU Emacs. ;; @@ -22,7 +18,6 @@ (require 'cl-lib) -(with-temp-buffer (insert "$ cd / $ ls dir a @@ -46,24 +41,47 @@ $ ls 8033020 d.log 5626152 d.ext 7214296 k") - (let ((data (split-string (buffer-string) "\n"))) - (cl-labels ((scan () - (cl-loop for item = (pop data) - while (and item (not (string= "$ cd .." item))) - when (parse item) - collect it)) - (parse (entry) - (pcase (split-string entry) - (`("$" "cd" ,dir) (cons dir (scan))) - (`("$" "ls") nil) - (`("dir" ,dir) nil) - (`(,size ,name) (cons name (string-to-number size)))))) - (scan) - ))) +(with-temp-buffer + (insert-file-contents "input") + (solver-add-dir-size + (solver-parse-listing (split-string (buffer-string) "\n")))) + +(defun solver-parse-listing (lines) + (cl-labels ((scan () + (cl-loop for item = (pop lines) + while (and item (not (string= "$ cd .." item))) + when (parse item) + collect it)) + (parse (entry) + (pcase (split-string entry) + (`("$" "cd" ,dir) (cons dir (scan))) + (`("$" "ls") nil) + (`("dir" ,dir) nil) + (`(,size ,name) (cons name (string-to-number size)))))) + + (car (scan)))) +(defun solver-add-dir-size (tree) + (cl-labels ((combiner (tri) + (list (car tri) + (adder tri) + (seq-reduce (lambda (acc item) + (if (numberp (cdr item)) + acc + (cons (combiner item) acc))) + (cdr tri) nil))) + (adder (tri) + (seq-reduce (lambda (acc item) + (+ acc + (if (numberp (cdr item)) + (cdr item) + (adder item)))) + (cdr tri) 0))) + (cl-loop for (_ size) on (flatten-tree (combiner tree)) by #'cddr + when (< size 100000) + sum size))) -(append '(5 6 8) (cons 5 8)) (provide 'solver) ;;; solver.el ends here |