Day 13

3 files changed, 559 insertions, 0 deletions

diff --git a/day13/__init__.py b/day13/__init__.py
new file mode 100644
index 0000000..d9fd9f2
--- /dev/null
+++ b/day13/__init__.py
@@ -0,0 +1,87 @@
+# -*- coding: utf-8 -*-
+import json
+from abc import ABC
+from functools import cmp_to_key, reduce
+from typing import Iterator, Any, List, Tuple, Union
+
+from aoc import BaseAssignment
+
+
+Packet = Union[List[int], int, List["Packet"]]
+
+
+class Assignment(BaseAssignment, ABC):
+    def compare_pair(self, pair: Tuple[Packet, Packet]):
+        packet_a, packet_b = pair
+
+        if isinstance(packet_a, int) and isinstance(packet_b, int):
+            diff = packet_b - packet_a
+            return diff / abs(diff) if diff != 0 else 0
+
+        if isinstance(packet_a, list) and isinstance(packet_b, int):
+            packet_b = [packet_b]
+
+        if isinstance(packet_b, list) and isinstance(packet_a, int):
+            packet_a = [packet_a]
+
+        try:
+            for sub_pair in zip(packet_a, packet_b, strict=True):
+                comparison = self.compare_pair(sub_pair)
+
+                if comparison != 0:
+                    return comparison
+        except ValueError:
+            return self.compare_pair((len(packet_a), len(packet_b)))
+
+        return 0
+
+
+class AssignmentOne(Assignment):
+    example_result = 13
+
+    def pairs(self, input: Iterator[str]):
+        while True:
+            yield (json.loads(next(input)), json.loads(next(input)))
+
+            try:
+                next(input)
+            except StopIteration:
+                break
+
+    def run(self, input: Iterator) -> Any:
+        return sum(
+            index
+            for index, pair in enumerate(self.pairs(input), start=1)
+            if self.compare_pair(pair) > 0
+        )
+
+
+class AssignmentTwo(Assignment):
+    example_result = 140
+
+    def packets(self, input: Iterator[str]):
+        for item in input:
+            if item == "":
+                continue
+
+            yield json.loads(item)
+
+    def run(self, input: Iterator) -> Any:
+        packets = sorted(
+            [[[2]], [[6]], *self.packets(input)],
+            key=cmp_to_key(lambda a, b: self.compare_pair((a, b))),
+            reverse=True,
+        )
+
+        return reduce(
+            lambda o, i: o * i,
+            [
+                index
+                for index, packet in enumerate(packets, start=1)
+                if packet
+                in (
+                    [[2]],
+                    [[6]],
+                )
+            ],
+        )
diff --git a/day13/example.txt b/day13/example.txt
new file mode 100644
index 0000000..af73fbb
--- /dev/null
+++ b/day13/example.txt
@@ -0,0 +1,23 @@
+[1,1,3,1,1]
+[1,1,5,1,1]
+
+[[1],[2,3,4]]
+[[1],4]
+
+[9]
+[[8,7,6]]
+
+[[4,4],4,4]
+[[4,4],4,4,4]
+
+[7,7,7,7]
+[7,7,7]
+
+[]
+[3]
+
+[[[]]]
+[[]]
+
+[1,[2,[3,[4,[5,6,7]]]],8,9]
+[1,[2,[3,[4,[5,6,0]]]],8,9]
diff --git a/day13/input.txt b/day13/input.txt
new file mode 100644
index 0000000..3655093
--- /dev/null
+++ b/day13/input.txt
@@ -0,0 +1,449 @@
+[[[[8,6]],[8,[7],[6,7,6,2,4]],10,[[1,7,9],7,[7,9]]],[[4,[],[10,5],[5,4,7],5],8,9,[[5,3,3,6,9],[9,5,10],8],[[0,6,9],[8],4,6,8]]]
+[[[],3,[[10,6,9,6],[6,8,7],[1,2]],8]]
+
+[[4,[4],[[10,7,2],[1,6,5,7,4],[7,3,3,1,5],[]],1],[[],[[7,6,3]],5,5]]
+[[[10]]]
+
+[[[[7,9]],[[0,9]],0,[[2,1,1,2,9],4],[[5]]],[]]
+[[8]]
+
+[[9,[[5,4,5,1,10]]],[[[6,9],3,[0,2]],5,6,[3,4,[2,2],10],[10,5,9]],[[[10,3,8],[],[6,9]],4]]
+[[[4,[7],[1,1,4,3,8],[3],[]],[[5,10,2,1],9,[2],6,7],4,[],[[10],[6,3,1,7]]],[4,[6],[[6],[6,6]]],[[[0,2,5],[3,5,10,7],[10,7,2,9],[],[0]],[[9,9,8],[]],[10,[2,10,7,7,3],4],[2,7]]]
+
+[[[],2,2,9,[[9],[1,10,5,1]]],[[[],9,5,[8]],10,9,[[1,4,0],[5]]]]
+[[5],[3],[],[9,[[7,2,0,9,3],[6,8,6],[]]],[[[6],3],6,[[2,8],8]]]
+
+[[]]
+[[[5,4],5]]
+
+[[7,7,5]]
+[[[[4,7,3],[6,0],1,6],[6,1,4,5]],[]]
+
+[[[[4,7,0,9,8],10,7,2]],[6,6,9],[[2,10],8,9,1,[1,[0,4]]],[[[2,0,2],[]]]]
+[[[[]],[],3],[],[[[],[8]],[[4,7,3,7,10],3,5]],[[9,[10,5,8,9,1]],0],[[[6,7,5,4,10],4,6,0,[8,3,1,2,7]],5,[10,1,5],10,0]]
+
+[[],[[3,8,2]]]
+[[8,4],[5],[3]]
+
+[[0,1,5,[[],[6,6,8,0],[0,2,5,9,9],[3,2,7],[5,2]],10],[],[[8,8],[1,3],3,2],[[[4,0],[]],[[4,1,2],[2],6,9,6],1,10,[[7],[0,7],5]],[4]]
+[[[],[6,[5,9,6,0,8],2,9,[4,2]],7,[9,6,0,10]]]
+
+[[],[],[],[9,[[1],[6,3,4,9]],10,8]]
+[[1,[[2],10,[0,9,3]]]]
+
+[[0,1,4,4],[[7,[6],[2,5,1,3],5],[[8,3,7],[6],[],8],8,1,[9,0,[]]],[[0,[3,9,3,5],[2,0,9,0,7],6]],[[[3,3,1,10,9]],[[0,7,1],8,[10,0],8],[[9,3,5,4],5,[0],5],8,[[3,6,10],0,2,6]]]
+[[[8,[1,2,10,9],2,[8,4,4]],[3,[5,5]],4],[[],2,[],6,3],[[10,7],[],7,[8,6,[5],4,3],2]]
+
+[[[[],1,2],[[7,7,7],6,[1,7,1]]],[4,8],[]]
+[[3,1,[1,6,[9,9,5],5],[[1,7,0,6],[10,8,10],[3,8],[6,2,7,5]]]]
+
+[[1]]
+[[[3],[[9,4,10,10],[8,9,1],[],[0,3,10,5]],[8,[]]],[[[10,6,0],2,1],5,0,[7],[[9]]],[[[],[9,9,3,2,3],5]]]
+
+[[],[7,2],[5,[9,[]],1,[4,6,5],[[9,10,8,3],6]]]
+[[[[]],0,[1,8],[5,2,7,2],6],[],[[],[[2,1,3,9],6,9,5,3],2],[[[10],[0],[4,0,3,6,8]],4],[9]]
+
+[[],[],[],[]]
+[[],[[3,6],6,[6],[[3,10,4],6],[]]]
+
+[[8,7,10],[3],[9,[],[[5,1,6,10,4]],4,[[9,8,1,1],[0,0,2],1]]]
+[[[1,6,[4,5,3,1]]],[[],[[7,0,5,1,9],6,[0,9,4,3]]]]
+
+[[2,2,1,8],[],[[[10,4,10,6],[2,8,3,10,6],[]],1,6],[6,0]]
+[[2,3,[8,[2,7]],8],[4,[[6,6,4,2],4,5,[7,5]],8,1,[]],[7]]
+
+[[[],[1,[9,1,9],[8,8,8,5]],3,[4,9,4,[2,6,6]],[]],[[7,[1,9,6]],6,[[10,0,0],[4,7,1],4,9,7]]]
+[[[[],8,5,1],3,[[8,2,2],4,[2,1,2,3,7]],10,[2,[5,7,6],3,[9,6,5,0,9],4]],[],[9,3,2,7],[4,[[4,10],[8,3,8]],6,[],9]]
+
+[[[],[[],[5]],4],[[0,[],9,[8,5,3,2,2]],[8]],[[[2,5]],[10,8,[7,8,7,3]],5,[[2,10],3,8,0,[6,9,10]],5],[[0,[7],[]],[[3,9,4,1,6],8],[[4,4,7,2],8],[[3],[],3],7]]
+[[[2,0,[7,5]],[1,5,7,6,[]],9,[2,1,[6,0],3,[3]]]]
+
+[[1,3,10,[]],[],[7,[8,4],[[],0,[4,8],6],4,[[4,0],[5,0,8,3],[0,1],8,[2,9,7,6,5]]],[3,5],[9,[[8,3,6,0],[10,0,5],10],[[10,1,4,0],4,8,5],9,[7,[0],[7,9],7,2]]]
+[[2,9,0,[[5,8,10,10,5],8,4,[5,6,10]],10],[8,[0,[10],[6],4,[]],[[9,4,0],[],[2,10,3,3,4],[3]],1,10],[[[],0,7,3,3],9,[[5,3,0],5,[4,6,6,7]],[]],[6,[[3,1],10,[0,8],0],8,[4,[],[8,0,7,9]],7],[9,4,[10],[[2,10,7,6]],1]]
+
+[[[],8,9,2,[6,[0,2,10,2],[4,9,7,2,10]]],[9,[[3,4]],5,9]]
+[[9],[[],[1,6,[5,9,9,9,0]],6]]
+
+[[5,2,10],[2,[[],7,9,3],0,[7,[]]],[[9,[5,6,5,0],4,0,3],10,1,[]],[8,10,3,9]]
+[[[2,[3]],9]]
+
+[[],[6,[3,[],[]],8,[[7,7,8,2],4],0]]
+[[[[0,5,7]],[[]],7,[4,[9,0,5],7,3]],[5,[[0,8,10,0],3],[1,4],[10],[]]]
+
+[[1,5],[10,[[9,7,1]],0,[9,[6,3],[7],9,[]]]]
+[[5,[2,[4,0,4,7,8],[4]]],[8,[2,9],5],[],[4,2]]
+
+[[7,[[0,0,5,4]],4],[[[2,4]],[1,[3,2,8],[9,1,8,2,2],0,1],2],[[[4,7],3,3,9]]]
+[[],[[[8],0,10,[0,4]],2,[9,2]],[10,4],[2,[7],10]]
+
+[[[],10],[[6,7,[8,6,6],2],5,[9,[5,7,9,0],[10,6],7]],[[],[5],10,[[6],6,7,6,[6]]],[[]]]
+[[[[],[]],[]],[[[6],[9],[4,0,8,5]],[3],[[8,2,7,7]],[],[3,[5,1],[9],[7,1]]],[[7,[0],[],3],[2]],[[]]]
+
+[7,10,4,6]
+[7,10,4,6,9]
+
+[[9]]
+[[7,2,6],[[6,9,[]],[5,3,6],[2],[[0,7,1],[0],[2],[9,9,0,8,2],[9,5,7,6,8]]],[[8,6,10,0,8],[9,[3,10],4,[2,4,3,4,4],5]],[],[]]
+
+[[[3,8,[]],1,7],[[4],[10]],[[[2],8,8],[[1,0,6,5],4],6,[4]],[[[],[9,8,7,10,6],2,[1]],[[],9,3,[1,6,5],[]],6,4,9]]
+[[8,[2,[3,7,1,0]]],[8,4,[[9,4],[0,3,5,5]]],[8,[[2,10,4],7,5,[9,0,5,0,4]],3,9,[8,8,3,3,0]],[[9,4,[0],4,[6]],[[0,10,0],6,[9,4,10,0]]]]
+
+[]
+[[[[0,4,6,7],3],2],[5,[10,[8],[4,9,4],[],5],6,[[0,6],[4,7,7],[],10,[]]]]
+
+[[[3],8,10,[[8],2]],[9,8,10],[[[],[],3,[]],[],[1,9]]]
+[[2,0,0,9],[[],[[10,2],[]],[7,[1],4,[2,5,8,1,2],6]],[[8],[[9,3,3,8],2,[5,5,2,9],2],[5,[],6,[2,1,4,10,1],7]],[],[]]
+
+[[[[1],[7,1,9]],[[],0,[0,7]],[6],9,[]],[]]
+[[[[2,8,7,3],[4,2],10,[8,4],7]],[],[[7,[3,2]]],[[[8,7],8,10],0,2]]
+
+[[[1,[6,7,0,8]],6],[2],[6,7],[],[[[1,9,2,10],[9],[0,4,3,1],[6,10,2,4],[4,0,10,6]],[6,5],6,[5,10,[1,0,10,1,2],[5,10,5],9],[[3]]]]
+[[[[3,9,0,6,3],[]],1,7,4,[[9,8,5],1,6,[]]]]
+
+[[],[[[]],9,[[2,6,2,9,5],9,[6,6]],[[1,0,7],[4,4]]]]
+[[1,[[5]],1,0],[2,[[9,9,1,8],2,8,[0,5]],[]],[0,[5,8,4],[[6,2],[3,6,0,8],[4,2,5],[5,10,4,6,6],[5,5,6]],[7,2,10,[]]],[7,[]]]
+
+[[5,[4],[0,[5,3,2],[]],0],[3,[10,2,7,[]]]]
+[[],[4,9],[[[2],[0,10,6,1],1,[0],[9,4,8,1,9]],0,[[0],8,9,[1,1,10,10]],7],[[[3,10],10],10,2,[10,3]],[[5]]]
+
+[[4,[],[[0,3,5]],[[]],1],[9,[0,4,7],[3,[5,10,9,2],6,2,6]],[]]
+[[[1,4,[10,3,8],[7,7]],[1,[]],[[1,2,5]],9],[],[],[0,6,2,8]]
+
+[[[[5,9,4,1]]],[0,1,[],7],[[[]],[9,[2,5,10],5,1],[[]],[[9],[6,4,0,4,9],1],[8]],[[10,1,7],6,10,7,6]]
+[[[8,9,[0,9,2,5]],10]]
+
+[[9,[[4,8,2,0],[0,1,7,7],5,[4,3,0]],6,[1,[3,9],[3,10,10,3,1],6]]]
+[[],[5,[[4,0,9,5],1,[2,0,5,6],8,3],1,[[9,3,10,10],[],[10],7]]]
+
+[[[[4,7,4,1],8],1],[[8,0,0,5,[0]],2,[]],[[[5,8,6,10,0],[],7,[],1],[7,[7,8,0,2,5],9,[8,5]],[8,[7,4]]],[10,[9,3,[0,5],[2]]],[[6,3,6],6,[[7,1,10],5,4,[3,5,1,7,10],9],[1,5,[2],[5],[3,7,0]]]]
+[[],[[],10],[0,[],5],[2,[8,2,0],[3]]]