You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.4 KiB
Clojure
45 lines
1.4 KiB
Clojure
3 years ago
|
(require '[clojure.string :as str])
|
||
|
|
|
||
|
|
(defn pad-input-map [in-map pad-val]
|
||
|
|
(let [pad-row (repeat (+ 4 (count (first in-map))) pad-val)
|
||
|
|
pad-side (repeat 2 pad-val)]
|
||
|
|
(mapv vec
|
||
|
|
(concat
|
||
|
|
[pad-row pad-row]
|
||
|
|
(mapv #(concat pad-side % pad-side) in-map)
|
||
|
|
[pad-row pad-row]))))
|
||
|
|
|
||
|
|
(defn get-number [in-map y x]
|
||
|
|
(concat (subvec (get in-map (dec y)) (dec x) (+ 2 x))
|
||
|
|
(subvec (get in-map y) (dec x) (+ 2 x))
|
||
|
|
(subvec (get in-map (inc y)) (dec x) (+ 2 x))))
|
||
|
|
|
||
|
|
(defn get-new-pixel [enhance-algo in-map y x]
|
||
|
|
(->> (get-number in-map y x)
|
||
|
|
str/join
|
||
|
|
(#(Integer/parseInt % 2))
|
||
|
|
(get enhance-algo)
|
||
|
|
{\. 0 \# 1}))
|
||
|
|
|
||
|
|
(defn build-output-number-map [enhance-algo in-map]
|
||
|
|
(let [pad-val (if (= \# (first enhance-algo)) (first (first in-map)) 0)
|
||
|
|
padded-in-map (pad-input-map in-map pad-val)]
|
||
|
|
(for [y (range 1 (dec (count padded-in-map)))]
|
||
|
|
(for [x (range 1 (dec (count (first padded-in-map))))]
|
||
|
|
(get-new-pixel enhance-algo padded-in-map y x)
|
||
|
|
))))
|
||
|
|
|
||
|
|
(defn count-lit-pixels [in-map]
|
||
|
|
((frequencies (flatten in-map)) 1))
|
||
|
|
|
||
|
|
(let [[enhance-algo in-map]
|
||
|
|
(->> (slurp "./in")
|
||
|
|
str/split-lines
|
||
|
|
((juxt
|
||
|
|
first
|
||
|
|
(comp (partial mapv #(mapv {\. 0 \# 1} %))
|
||
|
|
(partial drop 2)))))
|
||
|
|
image-output (iterate (partial build-output-number-map enhance-algo) in-map)]
|
||
|
|
(println (count-lit-pixels (nth image-output 50))))
|
||
|
|
|