|
|
@@ -7,6 +7,15 @@ with open(os.path.join(os.path.dirname(__file__), "example.txt")) as example:
|
|
|
with open(os.path.join(os.path.dirname(__file__), "input.txt")) as example:
|
|
|
input_data = example.read().splitlines()
|
|
|
|
|
|
+def bench(part):
|
|
|
+ import time
|
|
|
+ def wrapper(*args, **kwargs):
|
|
|
+ start = time.perf_counter()
|
|
|
+ value = part(*args, **kwargs)
|
|
|
+ print(f"\tevaluation time: {time.perf_counter() - start} s")
|
|
|
+ return value
|
|
|
+
|
|
|
+ return wrapper
|
|
|
|
|
|
def part1(data):
|
|
|
grid = np.array([list(w) for w in data])
|
|
|
@@ -23,11 +32,12 @@ def part1(data):
|
|
|
|
|
|
print(f"Part 1: {sum=}")
|
|
|
|
|
|
-
|
|
|
+@bench
|
|
|
def part2(data):
|
|
|
grid = np.array([list(w) for w in data])
|
|
|
sum = 0
|
|
|
- grid = np.pad(grid, pad_width=((1, 1), (1, 1)), mode="constant", constant_values="_")
|
|
|
+ grid = np.where(grid == "@", 1, 0)
|
|
|
+ grid = np.pad(grid, pad_width=((1, 1), (1, 1)), mode="constant", constant_values=0)
|
|
|
H, W = grid.shape
|
|
|
can_remove = True
|
|
|
rounds = 0
|
|
|
@@ -35,9 +45,9 @@ def part2(data):
|
|
|
coords2remove = []
|
|
|
for i in range(H - 3 + 2):
|
|
|
for j in range(W - 3 + 2):
|
|
|
- if grid[i, j] == "@":
|
|
|
+ if grid[i, j] == 1:
|
|
|
subgrid = grid[i - 1 : i + 2, j - 1 : j + 2]
|
|
|
- count = np.sum(subgrid == "@")
|
|
|
+ count = np.sum(subgrid)
|
|
|
if count < 5:
|
|
|
sum += 1
|
|
|
coords2remove.append((i, j))
|
|
|
@@ -45,7 +55,7 @@ def part2(data):
|
|
|
can_remove = False
|
|
|
else:
|
|
|
rounds += 1
|
|
|
- grid[tuple(zip(*coords2remove))] = "."
|
|
|
+ grid[tuple(zip(*coords2remove))] = 0
|
|
|
|
|
|
print(f"Part 2: {sum=} after {rounds} rounds")
|
|
|
|
