mirror of
https://github.com/peter-tanner/advent-of-code-2020.git
synced 2024-11-30 10:50:17 +08:00
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
|
|
import itertools
|
|
from copy import deepcopy
|
|
|
|
ic = [ list(x) for x in open('input').read().splitlines() ]
|
|
|
|
def generate_plane(space):
|
|
plane = space[0]
|
|
empty = [ [ '.' for i in range(len(plane[0])) ] for j in range(len(plane)) ]
|
|
space.insert(0, empty)
|
|
space.append(empty)
|
|
return space
|
|
space = generate_plane([ic])
|
|
|
|
def expand_space(space):
|
|
space = generate_plane(space)
|
|
space_ = deepcopy(space)
|
|
empty = [ '.' for i in range(len(space[0][0])+4) ]
|
|
for k in range(len(space)-1):
|
|
for j in range(len(space[k])):
|
|
space_[k][j] = ['.','.'] + space[k][j] + ['.','.']
|
|
space_[k].insert(0,deepcopy(empty))
|
|
space_[k].insert(0,deepcopy(empty))
|
|
space_[k].append(deepcopy(empty))
|
|
space_[k].append(deepcopy(empty))
|
|
# print(empty)
|
|
# print(len(space_[k]))
|
|
return space_
|
|
|
|
def search(i,j,k,space):
|
|
# print('search',i,j,k)
|
|
c = 0
|
|
for k_ in range(k-1,k+2):
|
|
for j_ in range(j-1,j+2):
|
|
for i_ in range(i-1,i+2):
|
|
if (k_ >= 0 and k_ < len(space) and j_ >= 0 and j_ < len(space[0]) and i_ >= 0 and i_ < len(space[0][0])) and (k_ != k or j_ != j or i_ != i):
|
|
if space[k_][j_][i_] == '#':
|
|
c+=1
|
|
# c+=1
|
|
# print()
|
|
# print(c)
|
|
return c
|
|
|
|
def spaceviewer(space): # Take a look at the current state of the system
|
|
c = 0
|
|
for z in space:
|
|
print(c)
|
|
c+=1
|
|
[print(y) for y in z]
|
|
|
|
# spaceviewer(space)
|
|
spaceviewer(space)
|
|
|
|
# print(len(space_),len(space_[0]),len(space_[0][0]))
|
|
def update_state(c, space):
|
|
space = expand_space(space)
|
|
space_ = deepcopy(space)
|
|
change = False
|
|
for k in range(0,len(space_)):
|
|
for j in range(0,len(space_[0])):
|
|
for i in range(0,len(space_[0][0])):
|
|
adj = search(i,j,k,space)
|
|
if space[k][j][i] == '#' and adj not in [2,3]:
|
|
space_[k][j][i] = '.'
|
|
change = True
|
|
elif space[k][j][i] == '.' and adj == 3:
|
|
space_[k][j][i] = '#'
|
|
change = True
|
|
if change == True and c < 5:
|
|
c+=1
|
|
spaceviewer(space_)
|
|
space_ = update_state(c, space_)
|
|
return space_
|
|
|
|
space_ = update_state(0,space)
|
|
|
|
|
|
print(sum([ sum([ y.count('#') for y in z ]) for z in space_ ])) |