This commit is contained in:
Tyrel Souza 2022-12-16 01:19:30 -05:00
parent fda3e791f4
commit 139b6d7e12
1 changed files with 4 additions and 10 deletions

View File

@ -26,15 +26,10 @@ class Valve:
def parse(rows): def parse(rows):
valves = {} valves = {}
for row in rows: for row in rows:
try: left, right = row.split(" valve")
left,right = row.split(" valves ") right = right.replace("s ", "").lstrip()
except ValueError:
left,right = row.split(" valve ")
valve, rate = scanf.scanf("Valve %s has flow rate=%d; %*s %*s to", left) valve, rate = scanf.scanf("Valve %s has flow rate=%d; %*s %*s to", left)
if "," in right: tunnels = right.split(", ")
tunnels = right.split(", ")
else:
tunnels = [right,]
valves[valve] = Valve(label=valve,rate=rate,tunnels=tunnels) valves[valve] = Valve(label=valve,rate=rate,tunnels=tunnels)
for _,v in valves.items(): for _,v in valves.items():
@ -42,7 +37,6 @@ def parse(rows):
return valves return valves
def part1(rows, sample=False): def part1(rows, sample=False):
p1 = Part1(rows,sample, 30) p1 = Part1(rows,sample, 30)
p1.run() p1.run()
@ -82,7 +76,7 @@ class Part1:
def move(self, where): def move(self, where):
self.tick += 1 self.tick += 1
self.cur = self.valves[where] self.cur = self.valves[where]
print("\tMove to valve", where) print("\tMove to valve", where)
def open(self): def open(self):