diff --git a/day16.py b/day16.py index 05b4d2a..1ed734f 100644 --- a/day16.py +++ b/day16.py @@ -24,29 +24,6 @@ class Tunnel: return str(self) -def strip_zero_flow(root: Valve) -> None: - q = deque() - q.append(root) - v = set() - while q: - valve = q.popleft() - if valve.name in v: - continue - v.add(valve.name) - for tunnel in valve.tunnels.copy(): - if tunnel.target.flowrate > 0: - continue - q.append(tunnel.target) - valve.tunnels.remove(tunnel) - for c_tunnel in tunnel.target.tunnels.copy(): - if c_tunnel.target == valve: - continue - elif c_tunnel.target == tunnel.target: - tunnel.target.tunnels.add(Tunnel(valve, tunnel.length + c_tunnel.length)) - else: - valve.tunnels.add(Tunnel(c_tunnel.target, tunnel.length + c_tunnel.length)) - - def get_openable_valve_tunnels(valve: Valve, open_valves: set, time_remaining: int) -> set: tunnels = set() queue = deque()