Create a gist now

Instantly share code, notes, and snippets.

; updated for Clojure SVN 1309
(defn parse-grid [s]
(vec (map (fn [line] (vec (map #(Integer. %) (.split line ","))))
(.split s "\\s+"))))
(defn main [gridstr]
(let [costs (parse-grid gridstr)
size (count costs)]
(loop [min-sums (vec (replicate size (vec (replicate size nil))))
work-queue (conj clojure.lang.PersistentQueue/EMPTY [0 0])]
(if (seq work-queue)
(let [yx (peek work-queue)
nbs (filter (fn [nyx] (every? #(< -1 % size) nyx))
(map #(map + yx %) [[-1 0] [1 0] [0 -1] [0 1]]))
nbs-costs (seq (filter identity (map #(get-in min-sums %) nbs)))
newsum (+ (get-in costs yx)
(if nbs-costs (apply min nbs-costs) 0))
oldsum (get-in min-sums yx)]
(if (or (nil? oldsum) (< newsum oldsum))
(recur (assoc-in min-sums yx newsum)
(apply conj (pop work-queue) nbs))
(recur min-sums (pop work-queue))))
(peek (peek min-sums))))))
(def testgrid "131,673,234,103,18
201,96,342,965,150
630,803,746,422,111
537,699,497,121,956
805,732,524,37,331
")
;(prn (time (main testgrid)))
(prn (time (main (slurp "matrix.txt"))))
; updated for Clojure SVN 1309
(defn parse-grid [s]
(vec (map (fn [line] (vec (map #(Integer. %) (.split line ","))))
(.split s "\\s+"))))
(defn main [gridstr]
(let [costs (parse-grid gridstr)
size (count costs)
min-sums (vec (for [_ costs] (vec (for [_ costs] (agent nil)))))
done (java.util.concurrent.Semaphore. 0)
remaining (java.util.concurrent.atomic.AtomicInteger. 1)
calc (fn calc [oldsum yx]
(let [nbs (filter (fn [nyx] (every? #(< -1 % size) nyx))
(map #(map + yx %)[[-1 0] [1 0] [0 -1] [0 1]]))
nbs-costs (seq (filter identity (map #(deref (get-in min-sums %)) nbs)))
newsum (+ (get-in costs yx)
(if nbs-costs (apply min nbs-costs) 0))]
(send (agent nil) (fn [_]
(when (zero? (.decrementAndGet remaining))
(.release done))))
(if (or (nil? oldsum) (< newsum oldsum))
(do
(.addAndGet remaining (count nbs))
(doseq [nyx nbs]
(send (get-in min-sums nyx) calc nyx))
newsum)
oldsum)))]
(send (ffirst min-sums) calc [0 0])
(.acquire done)
@(peek (peek min-sums))))
(def testgrid "131,673,234,103,18
201,96,342,965,150
630,803,746,422,111
537,699,497,121,956
805,732,524,37,331
")
(prn (time (main testgrid)))
;(prn (time (main (slurp "matrix.txt"))))
; updated for Clojure SVN 1309
(defn parse-grid
"Take a string that contains lines of comma-separated numbers, and
return a vector of vectors of Integers"
[s]
(vec (map (fn [line] (vec (map #(Integer. %) (.split line ","))))
(.split s "\\s+"))))
(defn neighbor-seqs
"Returns a vector of vectors of lazy seqs representing a square
matrix with size rows. Each lazy seq contains [y x] coordinates of
this cell's four neighbors to the north, south, east, and west."
[size]
(let [delta [[-1 0] [1 0] [0 -1] [0 1]]]
(vec (for [y (range size)]
(vec (for [x (range size)]
(filter (fn [nyx] (every? #(< -1 % size) nyx))
(map #(map + [y x] %) delta))))))))
(defn min-sum
"Calculates a new best-known cost sum to reach the given cell, based
on its own cost plus the smallest cost sum of its neighbors.
get-sum must be a function that takes a single [y x] coordinate
vector and returns the current cost sum of that cell, or nil if its
not known."
[oldsum yx costs neighbors get-sum]
(let [neighbor-sums (filter identity (map get-sum (get-in neighbors yx)))
newsum (+ (get-in costs yx)
(if (seq neighbor-sums) (apply min neighbor-sums) 0))]
(if (or (nil? oldsum) (< newsum oldsum))
newsum
oldsum)))
(defn use-queue
"Updates the cost sum of the top-left cells, adds its neighbors to a
work queue, and then repeats the process for all cells in the queue.
When the work queue is empty, the correct minimal cost is in the
bottom-right cell. The cost sum for that cell is returned."
[costs size neighbors]
(loop [sums (vec (replicate size (vec (replicate size nil))))
work-queue (conj clojure.lang.PersistentQueue/EMPTY [0 0])]
(if (empty? work-queue)
(peek (peek sums))
(let [yx (peek work-queue)
oldsum (get-in sums yx)
newsum (min-sum oldsum yx costs neighbors #(get-in sums %))]
(if (== oldsum newsum)
(recur sums (pop work-queue))
(recur (assoc-in sums yx newsum)
(apply conj (pop work-queue) (get-in neighbors yx))))))))
(defn use-agents
"Sets up a grid of agents. Sets up a watcher for each agent that
sends update actions to its neighboring agents. A count of all
scheduled cell agents is kept so that when all agent updates are
complete, the 'done' semaphore is released.This allows the calling
thread to collect and return the bottom-right agent's value."
[costs size neighbors]
(let [sums (vec (for [_ costs] (vec (for [_ costs] (agent nil)))))
get-sum #(deref (get-in sums %))
done (java.util.concurrent.Semaphore. 0)
remaining (java.util.concurrent.atomic.AtomicInteger. 1)]
(doseq [y (range size) x (range size)]
(add-watch (get-in sums [y x]) nil
(fn [agt _ old-val new-val]
(when (not= old-val new-val)
(let [nbs (get-in neighbors [y x])]
(.addAndGet remaining (count nbs))
(doseq [nyx nbs]
(send (get-in sums nyx) min-sum nyx
costs neighbors get-sum))))
(when (zero? (.decrementAndGet remaining))
(.release done)))))
(send (ffirst sums) min-sum [0 0] costs neighbors get-sum)
(.acquire done)
@(peek (peek sums))))
(defn main
"Solves PE #83 using either the use-queue or use-agent func above.
gridstr is the input grid given as described in parse-grid."
[func gridstr]
(let [costs (parse-grid gridstr)
size (count costs)
neighbors (neighbor-seqs size)]
(func costs size neighbors)))
(def testgrid "131,673,234,103,18
201,96,342,965,150
630,803,746,422,111
537,699,497,121,956
805,732,524,37,331
")
(def realgrid (slurp "matrix.txt"))
(prn (time (main use-queue realgrid)))
(prn (time (main use-agents realgrid)))
; answer is 425185
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment