blob: 7fbaf927d9aaa3c9c688544d76355d87d6bf323b (
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
|
(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]
(second
(first
(filter
#(-> (second %)
(str/replace (re-pattern (str/join ["[" dig-to-cmp "]"])) "")
(count)
(= segs-left)
)
(filterv #(= (first %) grp-to-cmp) cnts)
)
)
)
)
(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))
(assoc $ 2 (find-match 3 ($ 4) counts 5))
(assoc $ 5 (find-match 2 ($ 2) counts 5))
(assoc $ 0 (find-match 2 ($ 5) counts 6))
(assoc $ 9 (find-match 2 ($ 4) counts 6))
(assoc $ 6 (find-match 4 ($ 7) counts 6))
)
)
)
(loop [sum 0]
(let [rl (read-line)]
(if (empty? rl)
(println sum)
(let [line (as-> rl $
(str/split $ #" ")
(mapv (comp str/join sort) $)
)
number (subvec line 11 15)
decoder (set/map-invert (determine-digits line))]
(recur (->> number
(map (comp str decoder))
(str/join)
(#(Integer/parseInt %))
(+ sum)
)
)
)
)
)
)
|
