day23

day23.py

@@ -0,0 +1,132 @@
+input_test = '389125467'
+input = '476138259'
+
+moves = 10000000
+
+max_cups = 1000000
+cups = [int(c) for c in input] + [i for i in range(10, max_cups+1)]
+
+# part 1 structure
+# max_cups = 9
+# cups = [int(c) for c in input]
+
+
+# part one, with poor performance
+#
+# def calc_destination(destination, cups):
+#     global max_cups
+#     while destination not in cups:
+#         destination -= 1
+#         if destination < 1:
+#             destination = max_cups
+#     return destination
+#
+#
+# def calc_next_cup(current_cup, cups):
+#     new_current_cup_index = cups.index(current_cup)
+#     return (new_current_cup_index + 1) % len(cups)
+#
+#
+# def play(cups):
+#     current_cup_index = 0
+#     for m in range(moves):
+#         if m % 1000 == 0:
+#             print(m)
+#         current_cup = cups[current_cup_index]
+#         if current_cup_index + 4 < len(cups):
+#             pick = cups[current_cup_index+1:current_cup_index+4]
+#             cups = cups[:current_cup_index+1] + cups[current_cup_index+4:]
+#         else:
+#             pick = cups[current_cup_index+1:] + cups[:current_cup_index+4-len(cups)]
+#             cups = cups[current_cup_index+4-len(cups):current_cup_index+1]
+#
+#         destination = calc_destination(current_cup - 1, cups)
+#
+#         dest_index = cups.index(destination)
+#
+#         print("{}, {}, {}, {}".format(current_cup, current_cup_index, destination, pick))
+#
+#         cups = cups[:dest_index+1] + pick + cups[dest_index+1:]
+#
+#         current_cup_index = calc_next_cup(current_cup, cups)
+#     return cups
+#
+#
+# end = play(cups)
+#
+# index_one = end.index(1)
+#
+# final = end[index_one+1:] + end[:index_one]
+# print(''.join([str(i) for i in final]))
+
+def get_cup_dict(cups):
+    cup_dict = {}
+    for i, c in enumerate(cups):
+        if i == 0:
+            cup_dict[c] = (cups[-1], cups[i + 1])
+        elif i == len(cups) - 1:
+            cup_dict[c] = (cups[i-1], cups[0])
+        else:
+            cup_dict[c] = (cups[i-1], cups[i+1])
+    return cup_dict
+
+
+def calc_destination_simple(current_cup, not_available):
+    global max_cups
+    destination = current_cup-1
+    while destination in not_available or destination < 1:
+        if destination < 1:
+            destination = max_cups
+            continue
+        destination -= 1
+    return destination
+
+
+def play2(cups):
+    cup_dict = get_cup_dict(cups)
+    current_cup = cups[0]
+    for m in range(moves):
+        if m % 1000 == 0:
+            print(m)
+        current_cup_next_prev, first = cup_dict[current_cup]
+        first_prev, second = cup_dict[first]
+        _, third = cup_dict[second]
+        third_prev, fourth = cup_dict[third]
+
+        _, fourth_next = cup_dict[fourth]
+
+        cup_dict[fourth] = (current_cup, fourth_next)
+        cup_dict[current_cup] = (current_cup_next_prev, fourth)
+
+        pick = {first, second, third}
+        destination = calc_destination_simple(current_cup, pick)
+
+        destination_prev, destination_next = cup_dict[destination]
+
+        cup_dict[destination] = (destination_prev, first)
+        cup_dict[first] = (destination, second)
+        cup_dict[third] = (second, destination_next)
+
+        _, current_cup = cup_dict[current_cup]
+    return cup_dict
+
+
+end = play2(cups)
+
+_, first_after_one = end[1]
+_, second_after_one = end[first_after_one]
+
+print(first_after_one)
+print(second_after_one)
+print(first_after_one * second_after_one)
+
+# get result for part1
+#
+# _, current = end[1]
+#
+# res = ''
+# while current != 1:
+#     res += str(current)
+#     _, current = end[current]
+#
+# print(res)