day23

day23.py

@@ -0,0 +1,84 @@
+import aoclib
+
+DAY = 23
+TEST_SOLUTION_PART1 = "67384529"
+TEST_SOLUTION_PART2 = 149245887792
+
+
+class Cup:
+    def __init__(self, label, left=None, right=None):
+        self.label = label
+        self.left = left
+        self.right = right
+
+
+def buildDoubleLinkedListDict(cup_list):
+    cups = {}
+    left_cup = None
+    first_cup = None
+    last_cup = None
+    for cup in cup_list:
+        thisCup = Cup(cup, left_cup, None)
+        if first_cup is None:
+            first_cup = thisCup
+        else:
+            last_cup.right = thisCup
+        left_cup = thisCup
+        last_cup = thisCup
+        cups[cup] = thisCup
+
+    first_cup.left = last_cup
+    last_cup.right = first_cup
+
+    return cups, first_cup
+
+
+def play(cups, current_cup):
+    first_pickup = current_cup.right
+    first_after_pickup = current_cup.right.right.right.right
+    current_cup.right = first_after_pickup
+    first_after_pickup.left = current_cup
+
+    pickup_cups = [first_pickup.label, first_pickup.right.label, first_pickup.right.right.label]
+
+    current_label = current_cup.label - 1 if current_cup.label > 1 else max(cups.keys())
+    while current_label in pickup_cups:
+        current_label -= 1
+        if current_label == 0:
+            current_label = max(cups.keys())
+
+    desination_cup = cups[current_label]
+
+    desination_right = desination_cup.right
+    desination_cup.right = first_pickup
+    first_pickup.left = desination_cup
+    desination_right.left = first_pickup.right.right
+    first_pickup.right.right.right = desination_right
+
+    return cups, current_cup.right
+
+
+def part1(test_mode=False):
+    my_input = aoclib.getInputAs2DArray(day=DAY, return_type=int, test=test_mode)
+    cups, current_cup = buildDoubleLinkedListDict(my_input[0])
+
+    for round in range(100):
+        cups, current_cup = play(cups, current_cup)
+
+    answer = ""
+    add_cup = cups[1]
+    while add_cup.right.label != 1:
+        answer += str(add_cup.right.label)
+        add_cup = add_cup.right
+
+    return answer
+
+
+def part2(test_mode=False):
+    my_input = aoclib.getInputAs2DArray(day=DAY, return_type=int, test=test_mode)
+    cups, current_cup = buildDoubleLinkedListDict(my_input[0] + list(range(max(my_input[0]) + 1, 1000001)))
+
+    for round in range(10000000):
+        cups, current_cup = play(cups, current_cup)
+
+    return cups[1].right.label * cups[1].right.right.label

inputs/23

@@ -0,0 +1 @@
+792845136

inputs/23_test

@@ -0,0 +1 @@
+389125467