2022 Day 16part1 Python

2022/day16.py

@@ -41,12 +41,11 @@ def generate_path_costs(adjacent_valves: list[list]):
 	return path_costs
 
 T_MAX = 30
-times_called = np.zeros(T_MAX, dtype=dtype)
+times_called = np.zeros(T_MAX+1, dtype=dtype)
-def simulate(valves: dict):
+
-	# print(valves)
+
+def simulate(valves: dict, t_max=T_MAX):
 	v_keys = {k:i for i,k in enumerate(sorted(valves.keys()))}
-	# adjacent_valves = {k:v[1] for k,v in valves.items()}
-	# flows = {k:v[0] for k,v in valves.items()}
 	adjacent_valves = []
 	for k in v_keys:
 		adj = valves[k][1]
@@ -60,37 +59,36 @@ def simulate(valves: dict):
 	print(path_costs)
 
 	MAX_FLOW = sum(flows)
-	MAX_REMAINING = [i*MAX_FLOW for i in range(T_MAX, -1, -1)]
 
-	def sim_step(position, closed_valves, cur_flow=0, vented=0, t=0, max_vented=0) -> int:
+	def open_valve(valve: int, t: int) -> int:
+		return flows[valve] * (t_max - t)
+
+	def sim_step(position, closed_valves, t=0, vented=0, max_vented=0) -> int:
 		global times_called
 		times_called[t] += 1
 
-		t += 1
+		if t >= t_max:
-		# cur_flow = MAX_FLOW - sum((flows[i] for i in closed_valves))
+			print(f'sim_step finished with {position} {closed_valves}, {vented}, {t}')
-		vented += cur_flow
-		if t >= T_MAX:
-			# print(f'sim_step finished with {position} {closed_valves}, {vented}, {t}')
 			return max(max_vented, vented)
-		if (vented + MAX_REMAINING[t]) < max_vented:  # dead tree
+		# if (vented + (sum((flows[i] for i in closed_valves)) * (t_max - t))) < max_vented:  # dead tree
-			# print(f'sim_step died with {position} {closed_valves}, {vented}, {t}')
+		# 	print(f'sim_step died with {position} {closed_valves}, {vented}, {t}')
-			return max_vented
+		# 	return max(max_vented, vented)
 
-		if position in closed_valves:
-			max_vented = sim_step(position, closed_valves - {position}, cur_flow + flows[position], vented, t, max_vented)
 		for target_valve in closed_valves:
-			tn = min(t + path_costs[position,target_valve], T_MAX-1)
+			if target_valve == position:  # Open valve where we are
-			dt = tn - t - 1
+				tn = t+1
-			max_vented = sim_step(target_valve, closed_valves, cur_flow, vented+dt*cur_flow, tn, max_vented)
+				max_vented = sim_step(position, closed_valves - {position}, tn, vented + open_valve(target_valve, tn), max_vented)
-		if True:  #len(closed_valves) == 0:  # Wait it out
+			else:  # Teleport to next valve, pass time as if we walked there, and open it at that time
-			tn = T_MAX-1
+				tn = t + path_costs[position,target_valve] + 1
-			dt = tn - t - 1
+				if tn < t_max:
-			max_vented = sim_step(position, closed_valves, cur_flow, vented+dt*cur_flow, tn, max_vented)
+					max_vented = sim_step(target_valve, closed_valves - {target_valve}, tn, vented + open_valve(target_valve, tn), max_vented)
-		return max_vented
+		# print(f'sim_step unwound with {position} {closed_valves}, {vented}, {t}')
-
+		return max(vented, max_vented)
-	return sim_step(v_keys['AA'], {i for i,flow in enumerate(flows) if flow > 0})
+	default_closed = {i for i,flow in enumerate(flows) if flow > 0}
+	return sim_step(v_keys['AA'], default_closed)
 
 max_pressure_vented = simulate(parse(sample_lines))
 print(max_pressure_vented)
+print(times_called)
 max_pressure_vented = simulate(parse(lines))
 print(max_pressure_vented)

2022/input/16

@@ -0,0 +1,46 @@
+Valve OJ has flow rate=0; tunnels lead to valves EW, IG
+Valve BN has flow rate=0; tunnels lead to valves SA, AA
+Valve SA has flow rate=5; tunnels lead to valves QK, LP, ZP, BN, VH
+Valve RL has flow rate=21; tunnel leads to valve AM
+Valve LR has flow rate=19; tunnel leads to valve XZ
+Valve VQ has flow rate=0; tunnels lead to valves OW, IG
+Valve ZK has flow rate=0; tunnels lead to valves EW, WC
+Valve IG has flow rate=16; tunnels lead to valves OJ, VQ
+Valve WC has flow rate=22; tunnels lead to valves VD, ZK
+Valve EW has flow rate=18; tunnels lead to valves OJ, ZK
+Valve FP has flow rate=8; tunnel leads to valve GB
+Valve JF has flow rate=23; tunnel leads to valve VD
+Valve BL has flow rate=0; tunnels lead to valves AA, ZD
+Valve BZ has flow rate=0; tunnels lead to valves QK, JA
+Valve KH has flow rate=0; tunnels lead to valves SJ, FC
+Valve FU has flow rate=0; tunnels lead to valves FC, MH
+Valve ZP has flow rate=0; tunnels lead to valves SA, FC
+Valve DZ has flow rate=0; tunnels lead to valves AA, MH
+Valve RI has flow rate=0; tunnels lead to valves LP, MH
+Valve AE has flow rate=0; tunnels lead to valves FC, AA
+Valve JA has flow rate=4; tunnels lead to valves MM, BZ, JR, ZI, QO
+Valve XP has flow rate=0; tunnels lead to valves ZD, ZI
+Valve GB has flow rate=0; tunnels lead to valves FP, SJ
+Valve AM has flow rate=0; tunnels lead to valves ZD, RL
+Valve MH has flow rate=3; tunnels lead to valves VJ, DZ, JR, FU, RI
+Valve QK has flow rate=0; tunnels lead to valves BZ, SA
+Valve AA has flow rate=0; tunnels lead to valves DZ, CZ, BL, AE, BN
+Valve MJ has flow rate=0; tunnels lead to valves VN, VH
+Valve QO has flow rate=0; tunnels lead to valves CZ, JA
+Valve MM has flow rate=0; tunnels lead to valves FC, JA
+Valve VN has flow rate=17; tunnels lead to valves FV, MJ
+Valve OW has flow rate=0; tunnels lead to valves SJ, VQ
+Valve ZI has flow rate=0; tunnels lead to valves XP, JA
+Valve VJ has flow rate=0; tunnels lead to valves KJ, MH
+Valve KQ has flow rate=0; tunnels lead to valves XZ, KJ
+Valve FC has flow rate=12; tunnels lead to valves ZP, MM, KH, AE, FU
+Valve LP has flow rate=0; tunnels lead to valves SA, RI
+Valve VD has flow rate=0; tunnels lead to valves WC, JF
+Valve JR has flow rate=0; tunnels lead to valves MH, JA
+Valve VH has flow rate=0; tunnels lead to valves SA, MJ
+Valve CZ has flow rate=0; tunnels lead to valves AA, QO
+Valve SJ has flow rate=15; tunnels lead to valves KH, FV, GB, OW
+Valve FV has flow rate=0; tunnels lead to valves VN, SJ
+Valve XZ has flow rate=0; tunnels lead to valves LR, KQ
+Valve KJ has flow rate=9; tunnels lead to valves KQ, VJ
+Valve ZD has flow rate=13; tunnels lead to valves XP, BL, AM