44 lines
1.2 KiB
Clojure
44 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 ops (->> input
|
|
(drop-while num-line?)
|
|
(first)
|
|
(filter (partial not= \space))
|
|
(mapv {\+ +, \* *})))
|
|
|
|
;; 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]
|
|
(let [op (get ops op-idx)
|
|
ans (+ acc (apply op curr-nums))]
|
|
(if (> col-idx MAX_LINE_IDX)
|
|
ans
|
|
(let [col (->> nums-raw
|
|
(map #(.charAt % col-idx))
|
|
(filter (partial not= \space))
|
|
(map #(- (int %) (int \0))))]
|
|
(if (empty? col)
|
|
(recur (inc col-idx) (inc op-idx) [] ans)
|
|
(recur (inc col-idx)
|
|
op-idx
|
|
(conj curr-nums (parse-long (apply str col)))
|
|
acc))))))
|