blob: c5d2803daf2542d9a0876998aedb4269e669abfa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
(require '[clojure.core.reducers :as r])
(require '[clojure.string :as str])
(def input (->> (slurp "./in")
(str/split-lines)
(map
(fn [line]
[(if (str/starts-with? line "on") true false)
(->> (str/split line #"[^-\d]+")
(rest)
(map #(Integer/parseInt %))
(partition 2)
(mapv vec))]))
(reverse)))
(defn build-coord-list [coords idx]
(->> coords
(map #(update (get (second %) idx) 1 inc))
(flatten)
(sort)
(vec)))
(defn filter-coord-list [coords idx v]
(filter
#(let [[p0 p1] (get (second %) idx)]
(and (<= p0 v) (<= v p1)))
coords))
(def xs (build-coord-list input 0))
(def ys (build-coord-list input 1))
(def zs (build-coord-list input 2))
(defonce on-count (atom 0))
(loop [x xs]
(when-not (empty? (rest x))
(println "x=" (first x))
(let [inputx (filter-coord-list input 0 (first x))]
(swap! on-count +
(r/fold
32
+
(fn [onc2 iy]
(let [inputy (filter-coord-list inputx 1 (get ys iy))]
(+ onc2
(reduce
(fn [onc iz]
(let [inputz (first (filter-coord-list inputy 2 (get zs iz)))]
(cond-> onc
(first inputz)
(+ (* (- (second x) (first x))
(- (get ys (inc iy)) (get ys iy))
(- (get zs (inc iz)) (get zs iz)))))))
0
(range 0 (dec (count zs)))))))
(into [] (range 0 (dec (count ys))))))
(recur (rest x)))))
(println @on-count)
|
