programming-challenges/adventofcode2021/11-2.py at main · Ana06/programming-challenges · GitHub

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
import numpy as np
import sys


file_name = sys.argv[1] if len(sys.argv) > 1 else "input.txt"
locations = np.genfromtxt(file_name, delimiter=1, dtype=np.uint8)
shape = locations.shape
flashes = 0


def flash(i, j):
    global flashes
    if i < 0 or i >= shape[0] or j < 0 or j >= shape[1]:
        return
    if locations[i, j] != 0:
        locations[i, j] += 1
    if locations[i, j] > 9:
        locations[i, j] = 0
        flashes += 1
        flash(i - 1, j)
        flash(i + 1, j)
        flash(i, j - 1)
        flash(i, j + 1)
        flash(i - 1, j - 1)
        flash(i - 1, j + 1)
        flash(i + 1, j - 1)
        flash(i + 1, j + 1)


steps = 0
while True:
    steps += 1
    locations += 1
    previous_flashes = flashes
    while np.any(locations > 9):
        for i in range(shape[0]):
            for j in range(shape[1]):
                if locations[i, j] > 9:
                    flash(i, j)
    if flashes - previous_flashes == locations.size:
        print(steps)
        exit()