Day3 p2

3 files changed, 1108 insertions, 0 deletions

diff --git a/day3/__init__.py b/day3/__init__.py
new file mode 100644
index 0000000..29f4ab3
--- /dev/null
+++ b/day3/__init__.py
@@ -0,0 +1,96 @@
+from collections import Counter
+from copy import copy
+from dataclasses import dataclass
+from typing import Iterator, List
+
+from aoc import BaseAssignment
+
+class Assignment(BaseAssignment):
+    def read_input(self, example = False) -> List[str]:
+        return list(super().read_input(example))
+
+    @staticmethod
+    def flip(bits: str) -> str:
+        return bin(~int(bits, 2) + (1 << len(bits))).strip('0b')
+
+
+class AssignmentOne(Assignment):
+    example_result = 198
+    @staticmethod
+    def convert(input: List[str]) -> List[List[str]]:
+        items = []
+        for item in input:
+            for index, char in enumerate([char for char in item]):
+                try:
+                    items[index].append(char)
+                except IndexError:
+                    items.append([char])
+
+        return items
+
+    @staticmethod
+    def gamma_rate(input: List[str]) -> str:
+        converted = AssignmentOne.convert(input)
+        return ''.join([
+            Counter(i).most_common(1)[0][0]
+            for i
+            in converted
+        ])
+
+    @staticmethod
+    def epsilon_rate(input: List[str]) -> str:
+        return Assignment.flip(AssignmentOne.gamma_rate(input))
+
+    def run(self, input: List[str]) -> int:
+        return int(self.gamma_rate(input), 2) * int(self.epsilon_rate(input), 2)
+
+
+class AssignmentTwo(Assignment):
+    example_result = 230
+
+    @staticmethod
+    def oxygen_generator_rating(input: List[str]):
+        bit_length = len(input[0])
+
+        modified_input = copy(input)
+        bits = ''
+        for i in range(bit_length):
+            count = Counter([_[i] for _ in modified_input]).most_common(2)
+
+            most_used = count[0]
+            most_used_count = most_used[1]
+            least_used = count[-1]
+            least_used_count = least_used[1]
+
+            bit = '1' if most_used_count == least_used_count else most_used[0]
+
+            bits += bit
+            modified_input = list(filter(lambda _: _[i] == bit, modified_input))
+
+        return bits
+
+    @staticmethod
+    def co2_scrubbing_rating(input: List[str]):
+        bit_length = len(input[0])
+
+        modified_input = copy(input)
+        for i in range(bit_length):
+            if len(modified_input) == 1:
+                return modified_input[0]
+
+            count = Counter([_[i] for _ in modified_input]).most_common(2)
+
+            most_used = count[0]
+            most_used_count = most_used[1]
+            least_used = count[-1]
+            least_used_count = least_used[1]
+
+            bit = '0' if most_used_count == least_used_count else least_used[0]
+
+            modified_input = list(filter(lambda _: _[i] == bit, modified_input))
+
+
+    def run(self, input: List[str]) -> int:
+        return int(AssignmentTwo.oxygen_generator_rating(input), 2) \
+               * int(AssignmentTwo.co2_scrubbing_rating(input), 2)
+
diff --git a/day3/example.txt b/day3/example.txt
new file mode 100644
index 0000000..665fd57
--- /dev/null
+++ b/day3/example.txt
@@ -0,0 +1,12 @@
+00100
+11110
+10110
+10111
+10101
+01111
+00111
+11100
+10000
+11001
+00010
+01010
\ No newline at end of file
diff --git a/day3/input.txt b/day3/input.txt
new file mode 100644
index 0000000..16977ec
--- /dev/null
+++ b/day3/input.txt
@@ -0,0 +1,1000 @@
+000110000001
+011011001101
+001101100111
+001101011001
+110111011101
+110011101010
+111101010001
+010100111101
+011000011000
+001110110011
+001100010110
+110111101100
+110001111100
+001011111100
+000000011010
+110101100111
+011000011111
+011000000111
+011111000110
+100101110111
+010101001110
+111101000011
+010010010110
+100100011111
+101011001110
+001111110000
+110000011111
+110000011000
+011001111100
+010010001101
+000111001110
+110111001110
+110001010101
+100111011001
+000101110000
+110001011100
+111101010010
+101011000001
+001101001111
+110111101010
+101111111000
+110101000110
+011111001001
+001110100001
+010100110111
+110100000110
+101010110010
+100100101110
+101111011110
+000110110101
+011011110101
+111001011110
+110110100111
+000100010001
+001101010110
+100011000110
+001110010010
+010111110111
+011010011101
+110000011100
+010100001001
+000110100000
+101001010000
+000001110000
+101110010011
+010011100111
+010011011000
+110111011111
+000111010010
+101010111010
+111001100100
+101110100011
+111101111110
+010111111000
+010010001111
+110010000011
+001110000010
+100101111110
+000100101001
+101101010000
+111111010000
+101010011000
+011100100001
+011101101000
+001010010100
+010010100011
+110011111111
+110100011001
+111010011110
+011110001101
+011010011100
+100100000001
+111111001010
+110100110011
+110100011111
+100010001110
+101000111100
+100001110010
+110101010011
+101010011101
+011010011010
+101110101101
+001100011010
+101001010101
+101010000100
+000110101010
+100000111100
+000111111100
+000001001011
+010010111100
+011000111001
+111010101100
+010011100010
+010100110001
+001100111011
+101100001111
+111010101111
+010001000011
+000001100010
+000100001011
+100110011011
+101100001110
+000010110100
+011000101011
+010011011100
+110101010000
+101101101100
+101001110000
+010111010011
+110101110001
+011000001000
+011101010010
+111011111110
+010010000110
+110000111000
+101000000110
+011100110000