day4.py

#!/usr/bin/env/python

import re
import sys
from pathlib import Path

EXAMPLE_INPUT = """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
"""


def parse_input(i):
    for game_str in i.splitlines():
        prefix, winning_str, my_str = re.split("[:|]", game_str)
        winning = set(winning_str.split())
        my = set(my_str.split())
        yield (my, winning)


def get_scores(games):
    return [2 ** (n_win - 1) for winning, my in games if (n_win := len(winning & my))]


input_str = Path(sys.argv[1]).read_text() if len(sys.argv) > 1 else EXAMPLE_INPUT

print("Part 1", sum(get_scores(parse_input(input_str))))