Day 20 [WIP]

4 files changed, 5095 insertions, 0 deletions

diff --git a/day20/__init__.py b/day20/__init__.py
new file mode 100644
index 0000000..02ff80d
--- /dev/null
+++ b/day20/__init__.py
@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+from abc import ABC
+from collections import OrderedDict
+from typing import Iterator, List
+
+from aoc import BaseAssignment, I, T
+
+
+class Assignment(BaseAssignment[int, int], ABC):
+    def parse_item(self, item: str) -> int:
+        return int(item)
+
+
+class AssignmentOne(Assignment):
+    example_result = 3
+
+    @staticmethod
+    def move(items: List[int], item: int):
+        items = list(items)
+        input_size = len(items)
+
+        index = items.index(item)
+        item = items.pop(index)
+
+        new_index = (index + item) % input_size
+
+        if item > 0 and index + item >= input_size:
+            new_index += 1
+
+        if item < 0 and index + item <= 0:
+            new_index -= 1
+
+        items.insert(new_index, item)
+
+        return items
+
+    def run(self, input: Iterator[I]) -> T:
+        input_list = list(input)
+        input_size = len(input_list)
+        working_list = list(input_list)
+
+        for index, item in enumerate(input_list):
+            working_list = self.move(working_list, item)
+
+        print(
+            working_list[1000 % input_size],
+            working_list[2000 % input_size],
+            working_list[3000 % input_size],
+        )
+
+        return sum(
+            [
+                working_list[1000 % input_size],
+                working_list[2000 % input_size],
+                working_list[3000 % input_size],
+            ]
+        )
+
+
+class AssignmentTwo(Assignment):
+    pass
diff --git a/day20/example.txt b/day20/example.txt
new file mode 100644
index 0000000..52c68ff
--- /dev/null
+++ b/day20/example.txt
@@ -0,0 +1,7 @@
+1
+2
+-3
+3
+-2
+0
+4
diff --git a/day20/input.txt b/day20/input.txt
new file mode 100644
index 0000000..d79c61b
--- /dev/null
+++ b/day20/input.txt
@@ -0,0 +1,5000 @@
+-9405
+1055
+9806
+-52
+9756
+5575
+-6868
+9680
+-3207
+4001
+-8161
+-9566
+-1226
+1900
+2866
+4187
+1654
+3617
+8849
+-9914
+8521
+8120
+-5169
+-7619
+-7535
+7495
+5664
+-8496
+-5686
+-2536
+3805
+6796
+8547
+-1566
+1763
+185
+-8474
+-8685
+-620
+6756
+8177
+8512
+396
+-9955
+3824
+-944
+2517
+-8040
+5240
+3069
+-6780
+8507
+-441
+-5563
+-6580
+-871
+-3887
+-9017
+-3223
+8781
+-2683
+4879
+3489
+-2162
+5845
+-522
+-100
+-3908
+-3454
+9431
+1772
+8318
+6915
+-3412
+-8940
+3625
+-747
+-93
+7892
+1250
+-8964
+1441
+1461
+9017
+3088
+7405
+1726
+-4060
+6932
+4112
+4094
+-9703
+-2374
+6725
+-561
+-843
+-1936
+4973
+2838
+-6085
+-9957
+-6014
+-6600
+-1363
+2675
+8525
+9136
+-6674
+-6646
+-3119
+7283
+506
+-5932
+353
+9108
+-2531
+-8120
+-4082
+-3503
+5596
+-3053
+1505
+6886
+9246
+-3775
+3476
+6760
+-4143
+-7524
+398
+-4137
+162
+8094
+7805
+8702
+363
+7621
+-2204
+-5146
+6176
+-6085
+1571
+6284
+-8783
+-5776
+3058
+5006
+1497
+-6818
+-6673
+-4238
+-1158
+-4137
+5787
+6159
+-8251
+-7193
+-8504
+-2985
+7111
+-6426
+-609
+-6221
+641
+-2569
+-6092
+8744
+2435
+-5082
+4909
+-9216
+1782
+7040
+-7788
+-772
+8750
+7263
+-8535
+-4503
+-30
+6088
+5094
+-6322