application/octet-stream
•
2.88 KB
•
85 lines
#!/usr/bin/env nujel
(require :array/2d)
(def flashes/get)
(def flashes/inc!)
(let ((flashes/counter 0))
(set! flashes/get (fn () flashes/counter))
(set! flashes/inc! (fn () (inc! flashes/counter))))
(def p1-res 0)
(def p2-res 0)
(defn map/set/row (data y line w)
(def cols (map (split line "") read/int))
(dotimes (x w)
(array/2d/set! data x y (car cols))
(cdr! cols)))
(defn map/load (filename w h)
(def ret (array/2d/allocate w h))
(def lines (split (file/read filename) "\n"))
(dotimes (y h)
(map/set/row ret y (car lines) w)
(cdr! lines))
ret)
(defn step/increase-energy (α β)
(dotimes (x (ref α :width))
(dotimes (y (ref α :height))
(array/2d/set! β x y (+ 1 (array/2d/ref α x y))))))
(defn step/flash-increase (β flash-map x y)
(when (and (= #f (array/2d/ref flash-map x y))
(>= x 0)
(>= y 0)
(< x (ref β :width))
(< y (ref β :height)))
(array/2d/set! β x y (+ 1 (array/2d/ref β x y)))
(step/flash-try β flash-map x y)))
(defn step/flash-try (β flash-map x y)
(when-not (array/2d/ref flash-map x y)
(when (> (array/2d/ref β x y) 9)
(array/2d/set! β x y 0)
(array/2d/set! flash-map x y #t)
(flashes/inc!)
(def cx (- x 1))
(while (< cx (+ x 2))
(def cy (- y 1))
(while (< cy (+ y 2))
(step/flash-increase β flash-map cx cy)
(inc! cy))
(inc! cx)))))
(defn step/flash (β)
(def flash-map (array/2d/allocate (ref β :width) (ref β :height)))
(array/fill! (ref flash-map :data) #f)
(dotimes (x (ref β :width))
(dotimes (y (ref β :height))
(step/flash-try β flash-map x y))))
(defn step (α)
(def β (array/2d/allocate (ref α :width) (ref α :height)))
(step/increase-energy α β)
(step/flash β)
β)
(defn step/do-many (α steps)
(dotimes (i steps)
(if (= i 100) (set! p1-res (flashes/get)))
(def pre-flashes (flashes/get))
(set! α (step α))
(when (= 100 (- (flashes/get) pre-flashes))
(set! p2-res (+ 1 i))
(throw (list :all-flashes))))
α)
(def state (map/load "tests/slow/day11.dat" 10 10))
(try (fn (ε) (when (not= (car ε) :all-flashes) (throw ε)))
(def ret (step/do-many state 1000)))
(when (not= p1-res 1694)
(throw (list :wrong-result "Wrong result" p1-res)))
(when (not= p2-res 346)
(throw (list :wrong-result "Wrong result" p2-res)))
(return :success)
