I feel ashamed now because I could not understand problem correctly. And I basically copied pasted code from here and then refactored a little bit. So all the credit goes to the author.
Part 1
def part1():
with open("day17.txt") as fp:
lines = [fline.rstrip() for fline in fp.readlines()]
f = lines
cubes = {(i, j, 0):lines[i][j]
for i in range(len(lines))
for j in range(len(lines[0]))}
for i in range(6):
cubes = simulate(cubes)
print(f"Part 1 Solution: {list(cubes.values()).count('#')}")
def simulate(cubes):
new_cube = {}
for c in cubes:
x = checkNeighbors(c, cubes)
if cubes[c] == '#':
if x == 2 or x == 3:
new_cube[c] = '#'
else:
new_cube[c] = '.'
n = findNeighbors(c)
for x in n:
if x not in cubes:
k = checkNeighbors(x, cubes)
if k == 3:
new_cube[x] = '#'
elif cubes[c] == '.':
if x == 3:
new_cube[c] = '#'
else:
new_cube[c] = '.'
return new_cube
def findNeighbors(c):
neighbors = [(c[0]+i, c[1]+j, c[2]+k)
for i in range(-1, 2)
for j in range(-1, 2)
for k in range(-1, 2)
if not (i == 0 and j == 0 and k == 0)]
return neighbors
def checkNeighbors(c, cubes):
n = findNeighbors(c)
neighbors_count = len([x for x in n if cubes.get(x)=="#"])
return neighbors_count
part1()
def part2():
with open("day17.txt") as fp:
lines = [fline.rstrip() for fline in fp.readlines()]
f = lines
cubes = {(i, j, 0, 0):lines[i][j]
for i in range(len(lines))
for j in range(len(lines[0]))}
for i in range(6):
cubes = simulate4d(cubes)
print(f"Part 2 Solution: {list(cubes.values()).count('#')}")
def simulate4d(cubes):
new = {}
for c in cubes:
x = checkNeighbors4d(c, cubes)
if cubes[c] == '#':
if x == 2 or x == 3:
new[c] = '#'
else:
new[c] = '.'
n = findNeighbors4d(c)
for x in n:
if x not in cubes:
k = checkNeighbors4d(x, cubes)
if k == 3:
new[x] = '#'
elif cubes[c] == '.':
if x == 3:
new[c] = '#'
else:
new[c] = '.'
return new
def findNeighbors4d(c):
neighbors = [(c[0]+i, c[1]+j, c[2]+k, c[3]+w)
for i in range(-1, 2)
for j in range(-1, 2)
for k in range(-1, 2)
for w in range(-1, 2)
if not (i == 0 and j == 0 and k == 0 and w == 0)]
return neighbors
def checkNeighbors4d(c, cubes):
n = findNeighbors4d(c)
neighbors_count = len([x for x in n if cubes.get(x)=="#"])
return neighbors_count
part2()
Top comments (0)