feat: 2024 day 23

Author: Programator2

2024/23a.py

@@ -0,0 +1,29 @@
+from aoc import *
+from itertools import combinations
+from collections import defaultdict
+
+
+def main(infi: str):
+    sets = set()
+    connections = defaultdict(set)
+    computers = set()
+    for l in lines_stripped(infi):
+        left, right = l.split('-')
+        connections[left].add(right)
+        connections[right].add(left)
+        computers.add(left)
+        computers.add(right)
+    for a, b, c in combinations(computers, 3):
+        if b in connections[a] and b in connections[c] and c in connections[a]:
+            sets.add((a, b, c))
+    return sum(1 for a, b, c in sets if any(pc[0] == 't' for pc in (a, b, c)))
+
+
+DAY = 23
+FILE_TEST = f"{DAY}_testa.txt"
+# FILE_TEST = f"{DAY}_testb.txt"
+FILE_EXP = f"{DAY}_exp.txt"
+FILE = f"{DAY}.txt"
+# test_and_submit(main, FILE_TEST, FILE_EXP, FILE, DAY)
+# print(main(FILE_TEST))
+print(main(FILE))

2024/23b.py

@@ -0,0 +1,47 @@
+from aoc import *
+from functools import reduce
+from collections import defaultdict
+import operator
+
+
+def main(infi: str):
+    connections = defaultdict(set)
+    pairs = set()
+    networks = set()
+    for l in lines_stripped(infi):
+        left, right = l.split('-')
+        connections[left].add(right)
+        connections[right].add(left)
+        pairs.add(frozenset((left, right)))
+
+    done = set()
+
+    def create_networks(pair):
+        fpair = frozenset(pair)
+        if fpair in done:
+            return
+        done.add(fpair)
+
+        while newpcs := reduce(operator.and_, (connections[pc] for pc in pair)):
+            if len(newpcs) > 1:
+                for newpc in newpcs:
+                    create_networks(pair | set((newpc,)))
+                return
+            pair.update(newpcs)
+        networks.add(frozenset(pair))
+
+    for pair in pairs:
+        pair = set(pair)
+        create_networks(pair)
+
+    return ','.join(sorted(list(max(networks, key=lambda x: len(x)))))
+
+
+DAY = 23
+FILE_TEST = f"{DAY}_testa.txt"
+# FILE_TEST = f"{DAY}_testb.txt"
+FILE_EXP = f"{DAY}_exp.txt"
+FILE = f"{DAY}.txt"
+# test_and_submit(main, FILE_TEST, FILE_EXP, FILE, DAY)
+# print(main(FILE_TEST))
+print(main(FILE))

README.org

@@ -4,6 +4,7 @@
 * Personal Stats 2024
 | Day |     Time | Rank | Score |     Time |  Rank | Score |
 |-----+----------+------+-------+----------+-------+-------|
+|  23 | 00:24:26 | 2632 |     0 | 01:09:24 |  2938 |     0 |
 |  22 | 00:07:06 |  766 |     0 | 05:23:29 |  7143 |     0 |
 |  19 | 01:44:22 | 6641 |     0 | 01:50:06 |  5444 |     0 |
 |  17 | 00:27:08 | 1748 |     0 | 07:06:10 |  5688 |     0 |