public/aoc2020
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

day23: don't need left

Browse Source
This commit is contained in:
Stefan Harmuth 2020-12-23 11:41:36 +01:00
parent 5f0fb357bd
commit d82dcaa421
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
day23.py
View File

@ -6,9 +6,8 @@ TEST_SOLUTION_PART2 = 149245887792
class Cup: class Cup:
def __init__(self, label, left=None, right=None): def __init__(self, label, right=None):
self.label = label self.label = label
self.left = left
self.right = right self.right = right
@ -17,7 +16,7 @@ def buildDoubleLinkedListDict(cup_list):
left_cup = None left_cup = None
first_cup = None first_cup = None
for cup in cup_list: for cup in cup_list:
thisCup = Cup(cup, left_cup, None) thisCup = Cup(cup, None)
if first_cup is None: if first_cup is None:
first_cup = thisCup first_cup = thisCup
else: else:
@ -25,7 +24,6 @@ def buildDoubleLinkedListDict(cup_list):
left_cup = thisCup left_cup = thisCup
cups[cup] = thisCup cups[cup] = thisCup
first_cup.left = thisCup
thisCup.right = first_cup thisCup.right = first_cup
return cups, first_cup return cups, first_cup
@ -35,18 +33,14 @@ def play(cups, current_cup, max_label):
first_pickup = current_cup.right first_pickup = current_cup.right
first_after_pickup = current_cup.right.right.right.right first_after_pickup = current_cup.right.right.right.right
current_cup.right = first_after_pickup current_cup.right = first_after_pickup
first_after_pickup.left = current_cup
destination_label = current_cup.label - 1 if current_cup.label > 1 else max_label destination_label = current_cup.label - 1 if current_cup.label > 1 else max_label
while destination_label in [first_pickup.label, first_pickup.right.label, first_pickup.right.right.label]: while destination_label in [first_pickup.label, first_pickup.right.label, first_pickup.right.right.label]:
destination_label = destination_label - 1 if destination_label > 1 else max_label destination_label = destination_label - 1 if destination_label > 1 else max_label
desination_cup = cups[destination_label] desination_cup = cups[destination_label]
desination_right = desination_cup.right first_pickup.right.right.right = desination_cup.right
desination_cup.right = first_pickup 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 return cups, current_cup.right