|
| 1 | +(ns day25 |
| 2 | + (:require aoc)) |
| 3 | + |
| 4 | + |
| 5 | +(defn build-graph [components] |
| 6 | + (let [adjacencies (atom {})] |
| 7 | + (doseq [[k & vs] components |
| 8 | + v vs] |
| 9 | + (swap! adjacencies update k conj v) |
| 10 | + (swap! adjacencies update v conj k)) |
| 11 | + @adjacencies)) |
| 12 | + |
| 13 | + |
| 14 | +(defn add-edges [edge-counts vs] |
| 15 | + (reduce (fn [edge-counts [a b]] |
| 16 | + (update edge-counts (sort [a b]) (fnil inc 0))) |
| 17 | + edge-counts |
| 18 | + (partition 2 1 vs))) |
| 19 | + |
| 20 | +(defn traverse [graph start end edge-counts] |
| 21 | + (loop [queue (conj aoc/empty-queue [start []]) |
| 22 | + seen #{}] |
| 23 | + (if-let [[curr prevs] (peek queue)] |
| 24 | + (cond |
| 25 | + (= curr end) (add-edges edge-counts (conj prevs curr)) |
| 26 | + (seen curr) (recur (pop queue) seen) |
| 27 | + :else (recur (into (pop queue) (map (fn [x] [x (conj prevs curr)]) |
| 28 | + (graph curr))) |
| 29 | + (conj seen curr))) |
| 30 | + (count seen)))) |
| 31 | + |
| 32 | +(defn find-most-frequent-edges [graph] |
| 33 | + (reduce (fn [edge-counts [start end]] |
| 34 | + (traverse graph start end edge-counts)) |
| 35 | + {} |
| 36 | + (repeatedly 200 #(shuffle (keys graph))))) |
| 37 | + |
| 38 | +(defn remove-most-frequent [edge-counts graph] |
| 39 | + (let [most-frequent (keys (take 3 (sort-by val > edge-counts)))] |
| 40 | + (reduce (fn [graph [a b]] |
| 41 | + (-> graph |
| 42 | + (update a (fn [nbs] (remove #{b} nbs))) |
| 43 | + (update b (fn [nbs] (remove #{a} nbs))))) |
| 44 | + graph |
| 45 | + most-frequent))) |
| 46 | + |
| 47 | +(defn find-groups [graph] |
| 48 | + (let [edge-counts (find-most-frequent-edges graph) |
| 49 | + graph' (remove-most-frequent edge-counts graph) |
| 50 | + total-nodes (count (keys graph)) |
| 51 | + island (traverse graph' (key (first graph)) nil nil)] |
| 52 | + (* (- total-nodes island) island))) |
| 53 | + |
| 54 | + |
| 55 | +(defn solve [input-file] |
| 56 | + (let [components (aoc/parse-input input-file :words {:word-sep #": | "}) |
| 57 | + graph (build-graph components)] |
| 58 | + (find-groups graph))) |
| 59 | + |
| 60 | + |
| 61 | +(solve (aoc/read-file 25)) |
0 commit comments