2020-12-11 14:24:24 +01:00
|
|
|
from typing import Union
|
|
|
|
|
|
|
|
|
|
|
|
def parse_data() -> list[str]:
|
|
|
|
data = []
|
|
|
|
with open("input.txt") as file:
|
|
|
|
for line in file:
|
|
|
|
_line = line.rstrip()
|
|
|
|
data.append(_line)
|
|
|
|
return data
|
|
|
|
|
|
|
|
|
2020-12-11 18:10:06 +01:00
|
|
|
# Count occupied seats *adjacent* to the seat at position (x, y)
|
|
|
|
def count_occupied_adj(x: int, y: int, data: list[str]) -> int:
|
2020-12-11 14:24:24 +01:00
|
|
|
|
2020-12-11 18:10:06 +01:00
|
|
|
def check_occupied(_x: int, _y: int, _data: list[str]) -> int:
|
|
|
|
if 0 <= _x < len(data[0]) and 0 <= _y < len(data):
|
|
|
|
if data[_y][_x] == '#':
|
|
|
|
return 1
|
|
|
|
return 0
|
2020-12-11 14:24:24 +01:00
|
|
|
|
2020-12-11 18:10:06 +01:00
|
|
|
places = [(-1, -1), (-1, 0), (-1, 1), (0, -1), (0, 1), (1, -1), (1, 0), (1, 1)]
|
2020-12-11 14:24:24 +01:00
|
|
|
result = 0
|
2020-12-11 18:10:06 +01:00
|
|
|
for _x, _y in places:
|
|
|
|
result += check_occupied(x + _x, y + _y, data)
|
2020-12-11 14:24:24 +01:00
|
|
|
return result
|
|
|
|
|
|
|
|
|
|
|
|
def generate_new_grid(past_data: list[str]):
|
|
|
|
output = []
|
|
|
|
for y in range(len(past_data)):
|
|
|
|
line = ""
|
|
|
|
for x in range(len(past_data[0])):
|
2020-12-11 18:10:06 +01:00
|
|
|
if past_data[y][x] == "#" and count_occupied_adj(x, y, past_data) >= 4:
|
2020-12-11 14:24:24 +01:00
|
|
|
new_char = 'L'
|
2020-12-11 18:10:06 +01:00
|
|
|
elif past_data[y][x] == 'L' and count_occupied_adj(x, y, past_data) == 0:
|
2020-12-11 14:24:24 +01:00
|
|
|
new_char = '#'
|
|
|
|
else:
|
|
|
|
new_char = past_data[y][x]
|
|
|
|
line += new_char
|
|
|
|
output.append(line)
|
|
|
|
return output
|
|
|
|
|
|
|
|
|
|
|
|
def is_it_different(old_data: list[str], new_data: list[str]) -> bool:
|
|
|
|
for y in range(len(old_data)):
|
|
|
|
for x in range(len(old_data[0])):
|
|
|
|
if old_data[y][x] != new_data[y][x]:
|
|
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
def find_stable_state(grid: list[str], limit_iter: int = 10000) -> Union[None, list[str]]:
|
|
|
|
for i in range(limit_iter):
|
|
|
|
new_state = generate_new_grid(grid)
|
|
|
|
if not is_it_different(grid, new_state):
|
|
|
|
return new_state
|
|
|
|
grid = new_state
|
|
|
|
return None
|
|
|
|
|
2020-12-11 18:10:06 +01:00
|
|
|
|
2020-12-11 14:24:24 +01:00
|
|
|
def solve_p1(input_data: list[str]) -> int:
|
|
|
|
final_state = find_stable_state(input_data)
|
|
|
|
if final_state is None:
|
|
|
|
return -1
|
|
|
|
result = 0
|
2020-12-11 18:10:06 +01:00
|
|
|
for line in final_state:
|
|
|
|
result += line.count('#')
|
2020-12-11 14:24:24 +01:00
|
|
|
return result
|
|
|
|
|
|
|
|
|
2020-12-11 18:10:06 +01:00
|
|
|
print(solve_p1(parse_data()))
|