advent-of-code/2023/04/solver.py at main · chompaa/advent-of-code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import inspect
import os

cwd = os.path.dirname(os.path.abspath(inspect.stack()[0][1]))

with open(os.path.join(cwd, "example.txt"), "r") as f:
    cards = [
        (n1.split(": ")[-1].split(), n2.split())
        for n1, n2 in [card.split(" | ") for card in f.read().splitlines()]
    ]


def get_winning_amount(win, play):
    return sum(n in win for n in play)


# part 1

res = 0

for win, play in cards:
    amt = get_winning_amount(win, play)

    if amt != 0:
        res += 2 ** (amt - 1)

print(res)

# part 2

res = [1 for _ in range(len(cards))]

for num, (win, play) in enumerate(cards):
    amt = get_winning_amount(win, play)

    for i in range(num + 1, num + amt + 1):
        res[i] += res[num]

print(sum(res))