Day 4

3 files changed, 741 insertions, 0 deletions

diff --git a/day4/__init__.py b/day4/__init__.py
new file mode 100644
index 0000000..74c01f1
--- /dev/null
+++ b/day4/__init__.py
@@ -0,0 +1,121 @@
+from collections import Counter
+from copy import copy
+from dataclasses import dataclass
+from typing import Iterator, List, Optional, Tuple, Callable
+
+from aoc import BaseAssignment
+
+def bold(item: str, condition: Callable[[], bool]):
+    return f'\033[36m{item}\033[0m' if condition() else item
+
+@dataclass
+class Number:
+    nr: int
+    marked: bool = False
+
+    def __int__(self):
+        return self.nr
+
+    def __bool__(self):
+        return self.marked
+
+    def __radd__(self, other):
+        return other + int(self)
+
+    def mark(self):
+        self.marked = True
+
+@dataclass
+class BingoCard:
+    card: List[List[Number]]
+
+    @property
+    def bingo(self) -> bool:
+        return any([
+            *[all(bool(item) for item in row) for row in self.card],
+            *[all(bool(row[col_index]) for row in self.card) for col_index in range(5)]
+        ])
+
+    def mark(self, nr: int):
+        for row in self.card:
+            for item in row:
+                if int(item) == nr:
+                    item.mark()
+
+    @property
+    def unmarked(self):
+        return [
+            item
+            for row in self.card
+            for item in row
+            if not bool(item)
+        ]
+
+    def __str__(self):
+            return '\n'.join([
+                ''.join([
+                    bold(str(int(item)).rjust(2, ' ').ljust(3, ' '), lambda: item.marked)
+                    for item in row
+                ])
+                for row in self.card
+            ])
+
+
+class Assignment(BaseAssignment):
+    def read_input(self, example = False) -> Tuple[List[int], List[BingoCard]]:
+        cards = []
+        card = None
+
+        input = super().read_input(example)
+        nrs = [ int(n) for n in next(input).split(',') ]
+
+        try:
+            row = next(input)
+
+            while True:
+                if row == '':
+                    if card is not None:
+                        cards.append(BingoCard(card=card))
+                    card = []
+                else:
+                    card.append([ Number(nr=int(item)) for item in row.split(' ') if item != ''])
+
+                row = next(input)
+
+        except StopIteration:
+            if card is not None:
+                cards.append(BingoCard(card=card))
+
+        return nrs, cards
+
+
+class AssignmentOne(Assignment):
+    example_result = 4512
+
+    def run(self, input: Tuple[List[int], List[BingoCard]]) -> int:
+        nrs, cards = input
+
+        for nr in nrs:
+            for card in cards:
+                card.mark(nr)
+
+                if card.bingo:
+                    return nr * sum(card.unmarked)
+
+
+class AssignmentTwo(Assignment):
+    example_result = 1924
+
+    def run(self, input: Tuple[List[int], List[BingoCard]]) -> int:
+        nrs, cards = input
+
+        in_game = copy(cards)
+        for nr in nrs:
+            for card in copy(in_game):
+                card.mark(nr)
+
+                if card.bingo:
+                    in_game.remove(card)
+
+                if len(in_game) == 0:
+                    return nr * sum(card.unmarked)
\ No newline at end of file
diff --git a/day4/example.txt b/day4/example.txt
new file mode 100644
index 0000000..49d17bc
--- /dev/null
+++ b/day4/example.txt
@@ -0,0 +1,19 @@
+7,4,9,5,11,17,23,2,0,14,21,24,10,16,13,6,15,25,12,22,18,20,8,19,3,26,1
+
+22 13 17 11  0
+ 8  2 23  4 24
+21  9 14 16  7
+ 6 10  3 18  5
+ 1 12 20 15 19
+
+ 3 15  0  2 22
+ 9 18 13 17  5
+19  8  7 25 23
+20 11 10 24  4
+14 21 16 12  6
+
+14 21 17 24  4
+10 16 15  9 19
+18  8 23 26 20
+22 11 13  6  5
+ 2  0 12  3  7
\ No newline at end of file
diff --git a/day4/input.txt b/day4/input.txt
new file mode 100644
index 0000000..0c1c262
--- /dev/null
+++ b/day4/input.txt
@@ -0,0 +1,601 @@
+17,25,31,22,79,72,58,47,62,50,30,91,11,63,66,83,33,75,44,18,56,81,32,46,93,13,41,65,14,95,19,38,8,35,52,7,12,70,84,23,4,42,90,60,6,40,97,16,27,86,5,48,54,64,29,67,26,89,99,53,34,0,57,3,92,37,59,9,21,78,51,80,73,82,76,28,88,96,45,69,98,1,2,71,68,49,36,15,55,39,87,77,74,94,61,85,10,43,20,24
+
+36 11 70 77 80
+63  3 56 75 28
+89 91 27 33 82
+53 79 52 96 32
+58 14 78 65 38
+
+26 15 50 56  2
+20 27 42 11 16
+93 44 38 28 68
+66 88 78 81 77
+91 46 55 86  6
+
+46 53 14 17 75
+71  4 70 99 48
+65 96 68 80 72
+ 3 97 62 37 88
+82 35 36 23 39
+
+17  1 61 77  5
+74 60 12 24 48
+34 19 68 65 86
+44 59 38 40 95
+67 64  9 52 27
+
+44 60  8 81  3
+30 71 85 23 99
+68 88 38 97 48
+27 70 63 28 12
+67 57 34 13 93
+
+52 82 88 61  0
+68 21 59 75 71
+86 36 39 20 48
+50 40 19  6 34
+93 26 14 41 49
+
+74 18 93 59 77
+14 45 57 61 92
+10 78 42 63 52
+87  3  0 62 20
+25 64 48 22 96
+
+11 73 88 47 30
+10  6  5 25 67
+89 41 62 94 85
+45 99 58  7 57
+77 19 66 43 48
+
+94 47 45 73 44
+22  8 84 79  6
+14 58 26 92  5
+40 48 42 25  2
+37 76 18 80 74
+
+15  3 89 77 98
+13 99 10 97 59
+18 96 64 47 37
+68 92 90 56 11
+76 81 12 91 69
+
+82 97 69 47 10
+51 12 41 23 45
+71  6 67 80 46
+31 70 40  9 42
+27  1 17 25 74
+
+ 2 77  1 37 29
+50  8 87 12 76
+74 88 48 60 79
+41 35 92 33 34
+45 52 75 24 28
+
+97 41 49 40 96
+84 54 12 24 45
+39  1 17 85 52
+ 3 29 67 33  9
+50  7 47 48 81
+
+76 77 15 84 71
+41  7 32 29 62
+30 87 14 10 48
+98 22 96 45  9
+66 91 83 21 55
+
+20 42 33  9 91
+11 71 64 83 61
+82 54 67 38 60
+77 57 81 78 98
+18  1 27 55 87
+
+ 7 13 36 93 47
+45 25 44 58 72
+74 80 52 24 15
+64 43 96 42 20
+82 10 73 46 57
+
+40 36 86 87 76
+16 11 70 81 25
+55 31 83 72 88
+57 33 44 58  5
+64 15 19 67 53
+
+61 95 27  3 20