diff options
Diffstat (limited to '2023')
| -rw-r--r-- | 2023/04/solve.py | 46 |
1 files changed, 11 insertions, 35 deletions
diff --git a/2023/04/solve.py b/2023/04/solve.py index ea3c7fd..1391b06 100644 --- a/2023/04/solve.py +++ b/2023/04/solve.py @@ -1,43 +1,19 @@ -import re +L = [" ".join(l.split()) for l in open("input").readlines()] -L = [re.sub(" +", " ", l.strip()) for l in open("input").readlines()] - -def solve1(): - res = 0 - for l in L: - ls = l.split(": ") - card = ls[0] - winning, have = [ - list(map(int, a.split(" "))) for a in ls[1].split(" | ") - ] - s = 0 - i = True - for x in have: - if x in winning: - s = s + 1 if i else s * 2 - i = False - res += s - print(res) - - -def solve2(): - res = 0 - dup = [1] * len(L) +def solve(): + p1 = 0 + p2 = [1] * len(L) for i, l in enumerate(L): - ls = l.split(": ") - card = ls[0] + card, nums = l.split(": ") winning, have = [ - list(map(int, a.split(" "))) for a in ls[1].split(" | ") + list(map(int, a.split(" "))) for a in nums.split(" | ") ] - s = 0 - for x in have: - if x in winning: - s += 1 + s = sum(1 for x in have if x in winning) for j in range(i + 1, i + s + 1): - dup[j] += dup[i] - print(sum(dup)) + p2[j] += p2[i] + p1 += 0 if s == 0 else 2 ** (s - 1) + print(p1, sum(p2)) -solve1() -solve2() +solve() |