45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
|
def parse_tickets(lines):
|
||
|
|
tickets = []
|
||
|
|
for line in lines:
|
||
|
|
_, nums = line.rstrip().split(": ")
|
||
|
|
winning, played = nums.split(" | ")
|
||
|
|
winning, played = set(winning.split()), set(played.split())
|
||
|
|
tickets.append((winning, played))
|
||
|
|
return tickets
|
||
|
|
|
||
|
|
|
||
|
|
def part1(tickets):
|
||
|
|
total = 0
|
||
|
|
for ticket in tickets:
|
||
|
|
winning, played = ticket
|
||
|
|
num_wins = len(winning.intersection(played))
|
||
|
|
points = 0 if num_wins == 0 else 2**(num_wins-1)
|
||
|
|
total += points
|
||
|
|
print(f"part 1, total={total}")
|
||
|
|
|
||
|
|
|
||
|
|
def part2(tickets):
|
||
|
|
tickets = [[1, t] for t in tickets]
|
||
|
|
for index, ticket in enumerate(tickets):
|
||
|
|
mult = ticket[0]
|
||
|
|
winning, played = ticket[1]
|
||
|
|
num_wins = len(winning.intersection(played))
|
||
|
|
for i in range(index+1, index+1+num_wins):
|
||
|
|
tickets[i][0] += mult
|
||
|
|
num_tickets = sum(n for n, _ in tickets)
|
||
|
|
print(f"part 2, number of tickets: {num_tickets}")
|
||
|
|
|
||
|
|
|
||
|
|
def main(f):
|
||
|
|
tickets = parse_tickets(f)
|
||
|
|
part1(tickets)
|
||
|
|
part2(tickets)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import sys
|
||
|
|
infile = sys.argv[1]
|
||
|
|
with open(infile) as f:
|
||
|
|
main(f)
|
||
|
|
|