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
| import pytest
from pathlib import Path
def directions():
return [[-1,-1],[-1,0],[-1,1],[0,-1],[0,1],[1,-1],[1,0],[1,1]]
def get_raw_input():
return (Path(__file__).parent/'day_11_input.txt').read_text()
def parse_raw_input(raw_input):
return [[char for char in line.strip()] for line in raw_input.strip().splitlines()]
def get_adjacent(grid, seat, position):
if (y := seat[0]+position[0]) < 0:
return None
if (x := seat[1]+position[1]) < 0:
return None
try:
return grid[y][x]
except IndexError:
return None
def get_visible(grid, seat, direction):
y = seat[0]+direction[0]
x = seat[1]+direction[1]
try:
while y >= 0 and x >= 0 and grid[y][x] == '.':
y += direction[0]
x += direction[1]
if y >= 0 and x >= 0:
return grid[y][x]
else:
return None
except IndexError:
return None
def count_all_occupied_adjacent(grid, seat):
return len(
[
result for result
in (get_adjacent(grid, seat, position_to_check) for position_to_check in directions())
if result == '#'
]
)
def count_all_occupied_visible(grid, seat):
return len(
[
result for result
in (get_visible(grid, seat, direction_to_check) for direction_to_check in directions())
if result == '#'
]
)
def count_all_occupied(grid):
return len([seat for row in grid for seat in row if seat == '#'])
def tick(grid, method, trigger):
new_grid = [[seat for seat in row] for row in grid]
last_count = 0
iterations = 0
while last_count != count_all_occupied(new_grid) or last_count == 0:
last_count = count_all_occupied(grid)
for y, row in enumerate(grid):
for x, seat in enumerate(row):
current_seat = grid[y][x]
if method == 'visible':
count_of_occupied = count_all_occupied_visible(grid, [y, x])
else:
count_of_occupied = count_all_occupied_adjacent(grid, [y, x])
if current_seat == 'L' and count_of_occupied == 0:
new_grid[y][x] = '#'
elif current_seat == '#' and count_of_occupied >= trigger:
new_grid[y][x] = 'L'
grid = [[seat for seat in row] for row in new_grid]
return last_count
class TestExamples():
@pytest.fixture
def example_input(self):
return """L.LL.LL.LL
LLLLLLL.LL
L.L.L..L..
LLLL.LL.LL
L.LL.LL.LL
L.LLLLL.LL
..L.L.....
LLLLLLLLLL
L.LLLLLL.L
L.LLLLL.LL"""
def test_part_one_example(self, example_input):
assert tick(parse_raw_input(example_input), 'adjacent', 4) == 37
def test_part_two_example(self, example_input):
assert tick(parse_raw_input(example_input), 'visible', 5) == 26
if __name__ == '__main__':
print(tick(parse_raw_input(get_raw_input()), 'adjacent', 4))
print(tick(parse_raw_input(get_raw_input()), 'visible', 5))
|