day14: finished part1, part2 wip

1 files changed, 40 insertions, 0 deletions

diff --git a/day14/part2.clj b/day14/part2.clj
new file mode 100644
index 0000000..4e283ee
--- /dev/null
+++ b/day14/part2.clj
@@ -0,0 +1,40 @@
+(require '[clojure.string :as str])
+(require '[clojure.core.reducers :as r])
+
+(def input (->> (slurp "./in")
+                str/split-lines
+                ((juxt
+                   first
+                   (fn [lines]
+                     (->> lines
+                       (drop 2)
+                       (map #(str/split % #" -> "))
+                       (flatten)
+                       (apply (partial assoc {}))
+                       ))))))
+
+(defn handle-pair [pair steps]
+  (if (= 0 steps)
+    (frequencies (rest pair))
+    (let [ins ((second input) pair)
+          p1  (handle-pair (str/join [(first pair) ins]) (dec steps))
+          p2  (handle-pair (str/join [ins (second pair)]) (dec steps))]
+      (reduce
+        (fn [r p] (update r (key p) #(if (nil? %) (val p) (+ (val p) %))))
+        p1 p2)
+      )))
+
+(println
+  (reduce
+    (fn [tot pair]
+      (let [freqs (handle-pair pair 23)]
+        (println "Finished pair " pair)
+        (reduce
+          (fn [r p] (update r (key p) #(if (nil? %) (val p) (+ (val p) %))))
+          tot
+          freqs)))
+    {}
+    (for [i (range 0 (dec (count (first input))))]
+      (subs (first input) i (+ i 2)))
+    ))
+