85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
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
|