-
Notifications
You must be signed in to change notification settings - Fork 0
/
09.exs
185 lines (154 loc) · 4.76 KB
/
09.exs
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
defmodule Puzzle do
defmodule Group do
defstruct [:type, :level, content: ""]
end
def total_score(data) do
stats = stream_stats(data)
# stats |> IO.inspect(label: "Stats")
stats.groups
|> Enum.filter(&(&1.type == :plain))
|> Enum.map(&(&1.level))
|> Enum.sum
end
def garbage_chars(data) do
stats = stream_stats(data)
# stats |> IO.inspect(label: "Stats")
stats.groups
|> Enum.filter(&(&1.type == :garbage))
|> Enum.map(&(byte_size(&1.content)))
|> Enum.sum
end
@initial_stats %{stack: [], ignore: false, groups: []}
defp stream_stats(stream, stats \\ @initial_stats)
defp stream_stats("", stats), do: stats
defp stream_stats("!" <> rest, %{ignore: false} = stats) do
# IO.puts "Char: !"
stream_stats(rest, %{stats | ignore: true})
end
defp stream_stats(<<_char::utf8>> <> rest, %{ignore: true} = stats) do
# IO.puts "Ignoring char: #{<<char::utf8>>}"
stream_stats(rest, %{stats | ignore: false})
end
defp stream_stats(<<char::utf8>> <> rest, stats) do
# IO.puts "Char: #{<<char::utf8>>}"
{current, stack} = List.pop_at(stats.stack, 0)
stats =
if current && current.type == :garbage do
case char do
?> ->
# IO.puts "\tClosing garbage group!"
%{stats | stack: stack, groups: [current | stats.groups]}
_ ->
# IO.puts "\tContent: #{<<char::utf8>>}"
current = %{current | content: current.content <> <<char::utf8>>}
%{stats | stack: [current | stack]}
end
else
case char do
?{ ->
# IO.puts "\tNew group!"
group = %Group{type: :plain, level: next_level(current)}
%{stats | stack: [group | [current | stack]]}
?} ->
# IO.puts "\tClosing group!"
%{stats | stack: stack, groups: [current | stats.groups]}
?< ->
# IO.puts "\tNew garbage group!"
group = %Group{type: :garbage, level: next_level(current)}
%{stats | stack: [group | [current | stack]]}
_ ->
# IO.puts "\tContent: #{<<char::utf8>>}"
current = %{current | content: current.content <> <<char::utf8>>}
%{stats | stack: [current | stack]}
end
end
stream_stats(rest, stats)
end
defp next_level(nil), do: 1
defp next_level(current_group) do
current_group.level + 1
end
end
mode = List.first(System.argv)
if mode == "test" do
ExUnit.start()
defmodule PuzzleTest do
use ExUnit.Case
describe "total_score(data) for 1 group" do
test "returns expected" do
data = "{}"
score = Puzzle.total_score(data)
assert score == 1
end
end
describe "total_score(data) for 3 groups" do
test "returns expected" do
data = "{{{}}}"
score = Puzzle.total_score(data)
assert score == 6
end
end
describe "total_score(data) for 6 groups" do
test "returns expected" do
data = "{{{},{},{{}}}}"
score = Puzzle.total_score(data)
assert score == 16
end
end
describe "total_score(data) for 5 groups" do
test "returns expected" do
data = "{{<ab>},{<ab>},{<ab>},{<ab>}}"
score = Puzzle.total_score(data)
assert score == 9
end
end
describe "total_score(data) for 5 groups with !!" do
test "returns expected" do
data = "{{<!!>},{<!!>},{<!!>},{<!!>}}"
score = Puzzle.total_score(data)
assert score == 9
end
end
describe "total_score(data) for 2 groups" do
test "returns expected" do
data = "{{<a!>},{<a!>},{<a!>},{<ab>}}"
score = Puzzle.total_score(data)
assert score == 3
end
end
describe "garbage_chars(data) for 1 group" do
test "returns expected" do
data = "<>"
chars = Puzzle.garbage_chars(data)
assert chars == 0
end
end
describe "garbage_chars(data) for 1 < group" do
test "returns expected" do
data = "<<<<>"
chars = Puzzle.garbage_chars(data)
assert chars == 3
end
end
describe "garbage_chars(data) for 1 mixed group" do
test "returns expected" do
data = "<{o\"i!a,<{i<a>"
chars = Puzzle.garbage_chars(data)
assert chars == 10
end
end
describe "garbage_chars(data) for 1 big group" do
test "returns expected" do
data = "{{<a!>},{<a!>},{<a!>},{<ab>}}"
chars = Puzzle.garbage_chars(data)
assert chars == 17
end
end
end
else
data = "09.txt" |> File.read! |> String.trim_trailing
score = Puzzle.total_score(data)
IO.puts "The total score is: #{score}"
chars = Puzzle.garbage_chars(data)
IO.puts "The number of garbage chars is: #{chars}"
end