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
Row = Struct.new(:tiles) do
INPUT_MAP = { ?. => :safe, ?^ => :trap }
def self.from_s(input)
new(input.chars.map {|char| INPUT_MAP.fetch(char) })
end
def next
self.class.new(
%i[ safe safe ].insert(1, *tiles).each_cons(3).map {|l,c,r|
if l == :trap && c == :trap && r == :safe
:trap
elsif l == :safe && c == :trap && r == :trap
:trap
elsif l == :trap && c == :safe && r == :safe
:trap
elsif l == :safe && c == :safe && r == :trap
:trap
else
:safe
end
}
)
end
end
if __FILE__ == $0
row = Row.from_s(DATA.read.chomp)
count = 0
400_000.times do
count += row.tiles.count(:safe)
row = row.next
end
p count
end
require 'minitest'
# require 'minitest/autorun'
class TestRow < Minitest::Test
def test_row
row = Row.from_s('..^^.')
assert_equal %i[ safe safe trap trap safe ], row.tiles
assert_equal Row.from_s('.^^^^'), row.next
end
end
__END__