aoc/2023
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Further optimise day 4

Browse source
This commit is contained in:
Sijmen 2023-12-05 14:19:16 +01:00
parent 2194015259
commit 4dcc750416
1 changed files with 23 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
day04.py
View file

@ -1,13 +1,14 @@
import fileinput import fileinput
def main():
# #
# Part 1 # Part 1
# #
cards = [] cards = []
part1 = 0 part1 = 0
for i, line in enumerate(fileinput.input()): for line in fileinput.input():
card = line.rstrip().split(": ")[1] card = line.split(": ", 1)[1]
(winning, owned) = ({int(i) for i in h.split()} for h in card.split(" | ")) (winning, owned) = ({int(i) for i in h.split()} for h in card.split(" | ", 1))
count = len(winning & owned) count = len(winning & owned)
cards.append(count) cards.append(count)
part1 += 1 << count - 1 if count else 0 part1 += 1 << count - 1 if count else 0
@ -22,3 +23,5 @@ for i in range(len(cards)):
part2 += dp[i] part2 += dp[i]
print(part1, part2) print(part1, part2)
main()