You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

27 lines
699 B
Python

def do_task(lines):
win_sum = 0
for line in lines:
# print(line)
[wins, card] = line.split(": ")[1].split(" | ")
wins_set = set(wins.split())
card_set = set(card.split())
inter = wins_set.intersection(card_set)
# print(inter)
win_sum += 2 ** (len(inter) - 1) if len(inter) > 0 else 0
return win_sum
with open("day04/data/input100.txt") as infile:
lines = infile.readlines()
result = do_task(lines)
print(result)
print(result == 13)
with open("day04/data/input101.txt") as infile:
lines = infile.readlines()
result = do_task(lines)
print(result)
print(result == 22897)