-
Notifications
You must be signed in to change notification settings - Fork 0
/
adventofcode2021_day11.py
80 lines (63 loc) · 1.7 KB
/
adventofcode2021_day11.py
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
grid = []
def read_file(filename):
file = open(filename, "r")
lines = file.readlines()
grid.append([-1]*12)
for line in lines:
row = [-1]
for i in range(len(line.strip())):
row.append(int(line[i]))
row.append(-1)
grid.append(row)
grid.append([-1]*12)
def print_grid():
for row in grid:
for number in row:
if number != -1:
print(number, end="")
print()
def step():
#increment all by 1
for i in range(1, len(grid) - 1):
for j in range(1, len(grid) - 1):
grid[i][j] += 1
#flash all over 9
for i in range(1, len(grid) - 1):
for j in range(1, len(grid) - 1):
number = grid[i][j]
if number == 10:
flash(i, j)
#count number of flashes
flashes = 0
for row in grid:
for number in row:
if number == 0:
flashes += 1
return flashes
def flash(i, j):
grid[i][j] = 0
xadj = [-1, -1, -1, 0, 0, 1, 1, 1]
yadj = [-1, 0, 1, -1, 1, 1, 0, -1]
for x in range(len(xadj)):
number = grid[i + yadj[x]][j + xadj[x]]
if number > 0:
grid[i + yadj[x]][j + xadj[x]] += 1
if grid[i + yadj[x]][j + xadj[x]] > 9:
flash(i + yadj[x], j + xadj[x])
read_file("adventofcode2021_day11_input.txt")
# print_grid()
# step()
# flashes = step()
# print("Flashes: ", flashes)
# print_grid()
# step()
total_flashes = 0
allflashat = 0
for i in range(1000):
flashes = step()
total_flashes += flashes
if flashes == 100:
allflashat = i
break
print(total_flashes)
print("All flash at: ", allflashat)