27 lines
562 B
Python
27 lines
562 B
Python
import fileinput
|
|
|
|
def main():
|
|
#
|
|
# Part 1
|
|
#
|
|
cards = []
|
|
part1 = 0
|
|
for line in fileinput.input():
|
|
card = line.split(": ", 1)[1]
|
|
(winning, owned) = ({int(i) for i in h.split()} for h in card.split(" | ", 1))
|
|
count = len(winning & owned)
|
|
cards.append(count)
|
|
part1 += 1 << count - 1 if count else 0
|
|
|
|
#
|
|
# Part 2
|
|
#
|
|
dp = [0] * len(cards)
|
|
part2 = 0
|
|
for i in range(len(cards)):
|
|
dp[i] = 1 + sum(dp[i - cards[-i - 1] : i])
|
|
part2 += dp[i]
|
|
|
|
print(part1, part2)
|
|
|
|
main()
|