blob: af636832f8a99a64a34e6fe59d6a9670cbc20487 (
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
(require '[clojure.string :as str])
(defn read-coords []
(let [line (read-line)]
(when (not (empty? line))
(mapv
#(Integer/parseInt %)
(str/split
line
#"[^\d]+"
)
)
)
)
)
(defn mark-coord [cmap x y]
(vec
(for [c (range 0 (count cmap))]
(if (= c y)
(vec
(for [r (range 0 (count cmap))]
(if (= r x)
(inc (get (get cmap c) r))
(get (get cmap c) r)
)
)
)
(get cmap c)
)
)
)
)
(defn mark-coords [cmap x1 y1 x2 y2]
(cond
(= y1 y2)
(loop [cm cmap x (range (min x1 x2) (inc (max x1 x2)))]
(if (empty? x)
cm
(recur
(mark-coord cm (first x) y1)
(rest x)
)
)
)
(= x1 x2)
(loop [cm cmap y (range (min y1 y2) (inc (max y1 y2)))]
(if (empty? y)
cm
(recur
(mark-coord cm x1 (first y))
(rest y)
)
)
)
:else
(let [ic (if (< x1 x2) [x1 y1] [x2 y2])
ec (if (> x1 x2) [x1 y1] [x2 y2])
dy (if (> (second ec) (second ic)) 1 -1)
]
(loop [cm cmap c ic]
(if (> (first c) (first ec))
cm
(recur
(apply (partial mark-coord cm) c)
[(inc (first c)) (+ dy (second c))]
)
)
)
)
)
)
(defn empty-map []
(vec
(repeat 1000
(vec (repeat 1000 0))
)
)
)
(def finished-map
(loop [cmap (empty-map) coord (read-coords)]
(if (empty? coord)
cmap
(recur
(apply (partial mark-coords cmap) coord)
(read-coords)
)
)
)
)
(->> finished-map
(flatten)
(map dec)
(filter pos?)
(count)
(println)
)
|
