aoc 2024-day7

src/aoc/2024/day7.clj

@@ -0,0 +1,58 @@
+(ns aoc.2024.day7
+  (:require [util.core :as u]
+            [clojure.string :as s]))
+
+(defn parse [path]
+  (->> path
+       (u/read-input)))
+
+(defn prepare [path]
+  (->> path
+       (parse)
+       (map (fn [x] (s/split x #": ")))
+       (map (fn [[k v]] [(parse-long k) (map #(parse-long %) (s/split v #" "))]))))
+
+(defn process [repr]
+  (reduce (fn [{:keys [idx candi] :as state} val]
+            (if (empty? candi)
+              {:idx (inc idx) :candi (conj candi val)}
+              {:idx (inc idx) :candi (->> candi
+                                          (map (fn [x] [(+ x val) (* x val)]))
+                                          flatten)})) {:idx 0 :candi []} repr))
+
+(defn process2 [repr]
+  (reduce (fn [{:keys [idx candi] :as state} val]
+            (if (empty? candi)
+              {:idx (inc idx) :candi (conj candi val)}
+              {:idx (inc idx) :candi (->> candi
+                                          (map (fn [x] [(+ x val) (* x val) (parse-long (str x val))]))
+                                          flatten)})) {:idx 0 :candi []} repr))
+
+(defn calc [line process-fn]
+  (let [[goal repr] line
+        expected-results ((process-fn repr) :candi)]
+    {:result (some #(= %1 goal) expected-results)
+     :goal goal
+     :expected-results expected-results}))
+
+(defn part1 [path]
+  (->> path
+       prepare
+       (map #(calc % process))
+       (filter #(true? (:result %)))
+       (map :goal)
+       (reduce +)))
+
+(defn part2 [path]
+  (->> path
+       prepare
+       (map #(calc % process2))
+       (filter #(true? (:result %)))
+       (map :goal)
+       (reduce +)))
+
+(comment
+  (def path "resources/2024/input.txt")
+  (part1 path)
+  (part2 path)
+  )