AoC day 5 sketch

aoc5.clj

(require '[clojure.zip :as zip])
(def s '(0 3  0  1  -3))

(defn step [z]
  (let [v (zip/node z)
        dir (if (pos? v) zip/right zip/left)]
    (->> z
         (#(zip/edit % inc))
         (iterate dir)
         (take (inc (Math/abs v)))
         last)))
                 
(defn solve [s]
  (->> s
       zip/seq-zip
       zip/down
       (iterate step)
       (take-while some?)
       count))

(solve s)