feat(2023): day25

TurtleSmoke · Dec 25, 2023 · d52cfa4 · d52cfa4
1 parent a793166
commit d52cfa4
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 6 deletions.
diff --git a/2023/day_25/main.py b/2023/day_25/main.py
@@ -0,0 +1,24 @@
+#!/usr/bin/env python
+import itertools
+import math
+
+import networkx as nx
+
+input_file = "input"
+
+input_graph = nx.from_edgelist(
+    itertools.chain(
+        *(
+            itertools.product([src], dsts.split())
+            for src, dsts in [line.strip().split(":") for line in open(input_file)]
+        )
+    )
+)
+
+
+def split_graph(graph):
+    graph.remove_edges_from(nx.minimum_edge_cut(graph))
+    return nx.connected_components(graph)
+
+
+print(math.prod(map(len, split_graph(input_graph))))
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ readme = "README.md"
 packages = [{include = "advent_of_code"}]
 
 [tool.poetry.dependencies]
-python = "^3.8"
+python = "^3.10"
 numpy = "^1.23.5"
 pylint = "^2.15.8"
 black = "^22.10.0"
@@ -18,6 +18,7 @@ cpmpy = "^0.9.10"
 parse = "^1.20.0"
 sympy = "^1.12"
 z3-solver = "^4.12.4.0"
+networkx = "^3.2.1"
 
 [tool.black]
 line-length = 120