blob: 1ee3c0813876c9199a4afdeaab9349d4cbb2c187 (
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
61
62
63
64
65
66
|
(require '[clojure.string :as str])
(require '[clojure.set :as set])
(defn find-match
"
Searches for digit that `has segs-left` segments remaining
after the segments of `dig-to-cmp` are removed from the
digits in `cnts` with `grp-to-cmp` segments.
"
[segs-left dig-to-cmp cnts grp-to-cmp]
(as-> cnts $
(filterv #(= (first %) grp-to-cmp) $)
(filterv
#(-> (second %)
(str/replace (re-pattern (str/join ["[" dig-to-cmp "]"])) "")
(count)
(= segs-left)
)
$
)
(get-in $ [0 1])
)
)
(defn determine-digits [line]
(let [counts (mapv #(do [(count %) %]) (take 10 line))
mcounts (into {} counts)]
(as-> {} $
(assoc $ 1 (mcounts 2)
4 (mcounts 4)
7 (mcounts 3)
8 (mcounts 7)
)
(assoc $ 3 (find-match 2 ($ 7) counts 5)
6 (find-match 4 ($ 7) counts 6)
2 (find-match 3 ($ 4) counts 5)
9 (find-match 2 ($ 4) counts 6)
)
(assoc $ 5 (find-match 2 ($ 2) counts 5))
(assoc $ 0 (find-match 2 ($ 5) counts 6))
)
)
)
(println
(reduce
(fn [sum input]
(let [line (as-> input $
(str/split $ #" ")
(mapv (comp str/join sort) $)
)
number (subvec line 11 15)
decoder (set/map-invert (determine-digits line))]
(->> number
(map (comp str decoder))
(str/join)
(#(Integer/parseInt %))
(+ sum)
)
)
)
0
(str/split-lines (slurp "./in"))
)
)
|
