Completed 2023 Day 04

2023/python/Day04.py

+from math import log2
+from typing import List, Tuple
+
+from ImportData import import_data
+
+day: int = 4
+
+sample_data: List[str] = '''
+Card 1: 41 48 83 86 17 | 83 86  6 31 17  9 48 53
+Card 2: 13 32 20 16 61 | 61 30 68 82 17 32 24 19
+Card 3:  1 21 53 59 44 | 69 82 63 72 16 21 14  1
+Card 4: 41 92 73 84 69 | 59 84 76 51 58  5 54 83
+Card 5: 87 83 26 28 32 | 88 30 70 12 93 22 82 36
+Card 6: 31 18 13 56 72 | 74 77 10 23 35 67 36 11
+'''.split('\n')[1:-1]
+
+data_structure: type = List[Tuple[List[int], List[int]]]
+
+
+def parse_data(data: List[str]) -> data_structure:
+    cards: List[Tuple[List[int], List[int]]] = []
+    for card in data:
+        numbers: List[str] = card.split(':')[1].split('|')
+        winning_numbers: List[int] = [int(number_string) for number_string in numbers[0].strip().split()]
+        your_numbers: List[int] = [int(number_string) for number_string in numbers[1].strip().split()]
+        cards.append((winning_numbers, your_numbers))
+    return cards
+
+
+def get_score(card: Tuple[List[int], List[int]]) -> int:
+    score: int = 0
+    winning_numbers: List[int] = card[0]
+    your_numbers: List[int] = card[1]
+    for number in your_numbers:
+        if number in winning_numbers:
+            score = 1 if score == 0 else 2 * score
+    return score
+
+
+def part1(data: data_structure) -> int:
+    return sum([get_score(card) for card in data])
+
+
+def part2(data: data_structure) -> int:
+    copy_count: List[int] = [1 for _ in data]
+    number_of_cards = len(data)
+    for i in range(number_of_cards):
+        card: Tuple[List[int], List[int]] = data[i]
+        score: int = get_score(card)
+        if score > 0:
+            count: int = int(log2(score)) + 1  # kludgey
+            for j in range(i + 1, min(number_of_cards, i + count + 1)):
+                copy_count[j] += copy_count[i]
+    return sum(copy_count)
+
+
+if __name__ == '__main__':
+    production_ready = True
+    raw_data = import_data(day) if production_ready else sample_data
+    print(part1(parse_data(raw_data)))
+    print(part2(parse_data(raw_data)))