diff options
Diffstat (limited to 'day15/part1.clj')
-rw-r--r-- | day15/part1.clj | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/day15/part1.clj b/day15/part1.clj new file mode 100644 index 0000000..48dab23 --- /dev/null +++ b/day15/part1.clj @@ -0,0 +1,42 @@ +(def input (->> (slurp "./in") + (clojure.string/split-lines) + (mapv (partial mapv (comp read-string str))) + (#(for [y (range 0 (count %)) x (range 0 (count (first %)))] + {[y x] (get-in % [y x])})) + (into {}))) + +(defn min-distance [dist Q] + (reduce + #(if (and (not= (dist %2) :inf) + (or (= :inf (first %1)) + (< (dist %2) (first %1)))) + [(dist %2) %2] + %1) + [:inf []] Q)) + +(defn find-neighbors [Q u] + (filter #(contains? Q %) [(update u 0 inc) + (update u 0 dec) + (update u 1 inc) + (update u 1 dec)])) + +(let [dim (apply max (map first (keys input)))] + (loop [Q (set (keys input)) + dist (-> (zipmap Q (repeat :inf)) (update [0 0] #(do % 0)))] + (if (empty? Q) + (println (dist [dim dim])) + (let [[u vu] (min-distance dist Q)] + (recur + (disj Q vu) + (loop [d dist f (find-neighbors Q vu)] + (if (empty? f) + d + (recur + (let [v (first f) alt (+ u (input v))] + (if (or (= :inf (dist v)) (< alt (dist v))) + (assoc d v alt) + d + )) + (rest f) + )))))))) + |