blob: 8a57ec10233ae64089cc057885d91b72694b7ea4 (
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
|
(require '[clojure.string :as str])
(def caves (->> (slurp "./in")
(str/split-lines)
(map #(str/split % #"-"))
(map (partial map str))
(#(concat % (map reverse %)))
))
(def search-caves
(memoize (fn [id]
(filter #(= (first %) id) caves))))
(defn get-caves-forward [klst]
(let [end (first klst)]
(if (str/starts-with? end "end")
[klst]
(map #(cons (second %) klst)
(filter
(fn [cv]
(and
(not (str/starts-with? (second cv) "start"))
(or
(< (int (first (second cv))) 96)
(let [rv (count (filter #(= % (second cv)) klst))
lw (filter #(= (str/lower-case %) %) klst)]
(or (= 0 rv) (< (- (count lw) (count (distinct lw))) 1)))
)
))
(search-caves end)
)))))
(loop [lst (get-caves-forward ["start"]) ms []]
(println (count ms))
(let [nxt (->> lst
(map get-caves-forward)
(apply concat))
mtchs (concat ms (filter #(= (first %) "end") nxt))
nxtlst (filter #(not= (first %) "end") nxt)]
(if (empty? nxtlst)
(println (count mtchs))
(recur nxtlst mtchs))))
|
