2024, day 5, part 2 (too low)

Author: BioGeek

2024/day05.py

@@ -3,22 +3,29 @@
 
 def parse(protocol):
     top, bottom = protocol.split("\n\n")
-    ordering_rules = defaultdict(list)
+    ordering_rules = defaultdict(set)
     for line in top.splitlines():
         key, value = map(int, line.split("|"))
-        ordering_rules[key].append(value)
+        ordering_rules[key].add(value)
     updates = [list(map(int, line.split(","))) for line in bottom.splitlines()]
     return ordering_rules, updates
 
 
 def in_right_order(update, ordering_rules):
     for i in range(len(update)):
         number, *rest = update[i:]
-        if not set(rest).issubset(set(ordering_rules[number])):
+        if not set(rest).issubset(ordering_rules[number]):
             return False
     return True
 
 
+def order(update, ordering_rules):
+    print(update)
+    for n in update:
+        print(n, ordering_rules[n])
+    return sorted(update, key=lambda n: len(ordering_rules[n]), reverse=True)
+
+
 def middle_page_number(protocol, part=1):
     total = 0
     ordering_rules, updates = parse(protocol)
@@ -28,7 +35,9 @@ def middle_page_number(protocol, part=1):
             idx = len(update) // 2
             total += update[idx]
         if not is_in_right_order and part == 2:
-            pass
+            new_update = order(update, ordering_rules)
+            idx = len(new_update) // 2
+            total += new_update[idx]
     return total
 
 
@@ -71,3 +80,7 @@ def main(part: int = 1) -> int:
     assert middle_page_number(protocol) == 143
 
     print(main())
+
+    assert middle_page_number(protocol, part=2) == 123
+
+    print(main(part=2))