diff --git a/day11.py b/day11.py index 7b48fae..4998031 100644 --- a/day11.py +++ b/day11.py @@ -55,8 +55,10 @@ def simulate(layout, max_x, max_y, wide=False): occupied_seats += layout[(test_x, test_y)] break - if (layout[(x, y)] and occupied_seats >= max_seats_occupied) or (not layout[(x, y)] and occupied_seats == 0): - new_layout[(x, y)] = not layout[(x, y)] + if occupied_seats >= max_seats_occupied: + new_layout[(x, y)] = False + elif occupied_seats == 0: + new_layout[(x, y)] = True else: new_layout[(x, y)] = layout[(x, y)]