advent-of-code/2025/src/day06.clj
2025-12-05 20:36:52 -10:00

47 lines
1.2 KiB
Clojure

(ns day06
(:require input-manager))
(def input (input-manager/get-input 2025 6))
(def num-line? (comp not #{\* \+} first))
(def nums-raw (take-while num-line? input))
(def nums (map (comp (partial map parse-long)
#(re-seq #"\d+" %))
nums-raw))
(def op-map {\+ +, \* *})
(def ops (->> input
(drop-while num-line?)
(first)
(filter (partial not= \space))
(mapv op-map)))
;; part 1
(->> nums
(apply map (fn [op & rst]
(apply op rst))
ops)
(apply +))
;; part 2
(def MAX_LINE_IDX (dec (count (first nums-raw))))
(loop [col-idx 0
op-idx 0
curr-nums []
acc 0]
(if (> col-idx MAX_LINE_IDX)
(+ acc (apply (get ops op-idx) curr-nums))
(let [col (->> nums-raw
(map #(.charAt % col-idx))
(filter (partial not= \space))
(map #(- (int %) (int \0))))
op (get ops op-idx)]
(if (empty? col)
(recur (inc col-idx)
(inc op-idx)
[]
(+ acc (apply op curr-nums)))
(recur (inc col-idx)
op-idx
(conj curr-nums (parse-long (apply str col)))
acc)))))