Day 15 part 2
This commit is contained in:
parent
7b992a8bd7
commit
6677498314
|
@ -1,4 +1,5 @@
|
|||
from collections import defaultdict
|
||||
import time
|
||||
|
||||
from intcode import Intcode
|
||||
|
||||
|
@ -8,6 +9,7 @@ def main():
|
|||
text = input_file.readlines()[0].rstrip()
|
||||
codes = Intcode.parse_input(text)
|
||||
|
||||
# Part 1
|
||||
droid = Droid(codes)
|
||||
droid.run()
|
||||
print(f"Oxygen system at {droid.oxygen_system_pos}.")
|
||||
|
@ -15,12 +17,37 @@ def main():
|
|||
num_steps = len_backtrack(parents, droid.oxygen_system_pos, (0, 0))
|
||||
print(f"{num_steps} steps with BFS.")
|
||||
|
||||
# Part 2
|
||||
os_pos = droid.oxygen_system_pos
|
||||
droid.tiles[os_pos[0]][os_pos[1]] = Droid.TILE_O2
|
||||
minutes = 0
|
||||
propagations = [os_pos]
|
||||
while propagations:
|
||||
next_propagations = []
|
||||
for prop in propagations:
|
||||
adjacents = list(filter(
|
||||
lambda p: droid.tiles[p[0]][p[1]] == Droid.TILE_CLR,
|
||||
[Droid.forward(prop, d[1]) for d in Droid.NEXT_DIR.keys()]
|
||||
))
|
||||
for adj in adjacents:
|
||||
droid.tiles[adj[0]][adj[1]] = Droid.TILE_O2
|
||||
next_propagations += adjacents
|
||||
if not next_propagations:
|
||||
break
|
||||
propagations = next_propagations
|
||||
|
||||
droid.draw()
|
||||
minutes += 1
|
||||
print(f"Minute {minutes}: propagated to {propagations}")
|
||||
time.sleep(0.2)
|
||||
|
||||
|
||||
class Droid(Intcode):
|
||||
|
||||
TILE_UNK = 0
|
||||
TILE_WALL = 1
|
||||
TILE_CLR = 2
|
||||
TILE_O2 = 4
|
||||
|
||||
RES_COLL = 0
|
||||
RES_OK = 1
|
||||
|
@ -57,15 +84,14 @@ class Droid(Intcode):
|
|||
if data == Droid.RES_COLL:
|
||||
self.tiles[fx][fy] = Droid.TILE_WALL
|
||||
self.dir = Droid.NEXT_DIR[self.dir]
|
||||
elif data == Droid.RES_OK:
|
||||
self.halt = (fx, fy) == (0, 0)
|
||||
else:
|
||||
self.x, self.y = fx, fy
|
||||
self.tiles[fx][fy] = Droid.TILE_CLR
|
||||
self.dir = Droid.PREV_DIR[self.dir]
|
||||
elif data == Droid.RES_END:
|
||||
self.x, self.y = fx, fy
|
||||
self.tiles[self.x][self.y] = Droid.TILE_CLR
|
||||
if data == Droid.RES_END:
|
||||
self.oxygen_system_pos = self.x, self.y
|
||||
if self.halt or self.steps % 100 == 0:
|
||||
self.halt = (fx, fy) == (0, 0)
|
||||
if self.halt:
|
||||
self.draw()
|
||||
|
||||
def get_forward_pos(self):
|
||||
|
@ -79,8 +105,8 @@ class Droid(Intcode):
|
|||
for x in range(-24, 24):
|
||||
for y in range(-24, 24):
|
||||
if (x, y) == self.oxygen_system_pos:
|
||||
print("X", end="")
|
||||
if (x, y) == (0, 0):
|
||||
print("@", end="")
|
||||
elif (x, y) == (0, 0):
|
||||
print("s", end="")
|
||||
elif (x, y) == (self.x, self.y):
|
||||
print("D", end="")
|
||||
|
@ -89,6 +115,7 @@ class Droid(Intcode):
|
|||
Droid.TILE_UNK: " ",
|
||||
Droid.TILE_WALL: "█",
|
||||
Droid.TILE_CLR: "░",
|
||||
Droid.TILE_O2: "▒",
|
||||
}[self.tiles[x][y]], end="")
|
||||
print()
|
||||
|
||||
|
@ -102,11 +129,11 @@ def bfs(tiles, start, end):
|
|||
pos = q.pop(0)
|
||||
if pos == end:
|
||||
return parents
|
||||
adjacent = filter(
|
||||
adjacents = filter(
|
||||
lambda p: tiles[p[0]][p[1]] != Droid.TILE_WALL,
|
||||
[Droid.forward(pos, d[1]) for d in Droid.NEXT_DIR.keys()]
|
||||
)
|
||||
for adj in adjacent:
|
||||
for adj in adjacents:
|
||||
if discovered[adj[0]][adj[1]]:
|
||||
continue
|
||||
discovered[adj[0]][adj[1]] = True
|
||||
|
|
Loading…
Reference in a new issue