Jake Pullen
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
with open(r'advent_of_code\2023\19\input.txt', 'r') as file:
|
|
input = file.read()
|
|
|
|
test_input = '''px{a<2006:qkq,m>2090:A,rfg}
|
|
pv{a>1716:R,A}
|
|
lnx{m>1548:A,A}
|
|
rfg{s<537:gd,x>2440:R,A}
|
|
qs{s>3448:A,lnx}
|
|
qkq{x<1416:A,crn}
|
|
crn{x>2662:A,R}
|
|
in{s<1351:px,qqz}
|
|
qqz{s>2770:qs,m<1801:hdj,R}
|
|
gd{a>3333:R,R}
|
|
hdj{m>838:A,pv}
|
|
|
|
{x=787,m=2655,a=1222,s=2876}
|
|
{x=1679,m=44,a=2067,s=496}
|
|
{x=2036,m=264,a=79,s=2244}
|
|
{x=2461,m=1339,a=466,s=291}
|
|
{x=2127,m=1623,a=2188,s=1013}'''
|
|
|
|
# Split the input into two blocks
|
|
#input = test_input
|
|
block1, block2 = input.split("\n\n")
|
|
|
|
# Initialize a dictionary to store workflows
|
|
workflows_dict = {}
|
|
|
|
# Process the first block of input
|
|
for line in block1.splitlines():
|
|
workflow_name, remaining = line[:-1].split("{")
|
|
rules = remaining.split(",")
|
|
workflows_dict[workflow_name] = ([], rules.pop())
|
|
for rule in rules:
|
|
comparison, target = rule.split(":")
|
|
comparison_key = comparison[0]
|
|
comparison_operator = comparison[1]
|
|
comparison_value = int(comparison[2:])
|
|
workflows_dict[workflow_name][0].append((comparison_key, comparison_operator, comparison_value, target))
|
|
|
|
# Function to calculate product of ranges
|
|
def calculate_product(ranges):
|
|
product = 1
|
|
for low, high in ranges.values():
|
|
product *= high - low + 1
|
|
return product
|
|
|
|
# Function to count the number of accepted items
|
|
def count_accepted_items(ranges, workflow_name = "in"):
|
|
if workflow_name == "R":
|
|
return 0
|
|
if workflow_name == "A":
|
|
return calculate_product(ranges)
|
|
|
|
rules, fallback = workflows_dict[workflow_name]
|
|
|
|
total = 0
|
|
|
|
for comparison_key, comparison_operator, comparison_value, target in rules:
|
|
low, high = ranges[comparison_key]
|
|
if comparison_operator == "<":
|
|
true_range = (low, comparison_value - 1)
|
|
false_range = (comparison_value, high)
|
|
else:
|
|
true_range = (comparison_value + 1, high)
|
|
false_range = (low, comparison_value)
|
|
if true_range[0] <= true_range[1]:
|
|
copy_ranges = dict(ranges)
|
|
copy_ranges[comparison_key] = true_range
|
|
total += count_accepted_items(copy_ranges, target)
|
|
if false_range[0] <= false_range[1]:
|
|
ranges = dict(ranges)
|
|
ranges[comparison_key] = false_range
|
|
else:
|
|
break
|
|
else:
|
|
total += count_accepted_items(ranges, fallback)
|
|
|
|
return total
|
|
|
|
# Print the total count
|
|
print(count_accepted_items({key: (1, 4000) for key in "xmas"})) |