advent-of-code/2024-python/output/day_24.py
2025-01-05 00:06:18 +01:00

72 lines
1.8 KiB
Python

import graphviz
import re
def solve(data):
values, wires = data.split("\n\n")
values = dict([tuple(c.strip().split(": ")) for c in values.splitlines()])
wires = [w.split() for w in wires.splitlines()]
Q = [*wires]
while Q:
a, o, b, _, c = Q.pop(0)
if a not in values or b not in values:
Q.append((a, o, b, "_", c))
continue
a, b = int(values[a]), int(values[b])
match o:
case "AND":
values[c] = int(a + b == 2)
case "XOR":
values[c] = int(a + b == 1)
case "OR":
values[c] = int(a + b > 0)
p1 = int(
"".join(
[
str(v)
for k, v in sorted(values.items(), key=lambda x: x[0], reverse=True)
if k.startswith("z")
]
),
2,
)
# render_graph(wires)
wrong_xors_inputs = r"[^y|x]\S{2}+ XOR [^x|y]\S{2}+ -> ([^z]\S{2})"
wrong_z_outputs = r"\s(?:OR|AND) .+ -> (z\d{2})"
found_manually = ["jqf", "skh"]
p2 = ",".join(
sorted(
[seq for seq in re.findall(wrong_z_outputs, data) if not seq.endswith("45")]
+ list(re.findall(wrong_xors_inputs, data))
+ found_manually
)
)
return p1, p2
def render_graph(wires):
dot = graphviz.Digraph(comment="The Round Table")
for a, o, b, _, c in wires:
ol = f"{a}{b}{o}{c}"
dot.node(a)
dot.node(b)
dot.node(c)
dot.node(ol, label=o, shape="diamond")
dot.edge(a, ol)
dot.edge(b, ol)
dot.edge(ol, c)
dot.render("2024.24.p2.gv").replace("\\", "/")
if __name__ == "__main__":
with open("./input/24.txt", "r") as f:
inp = f.read().strip()
p1, p2 = solve(inp)
print(p1)
print(p2)