Day 19 Done

2023/19/part 1 solution.py

@@ -0,0 +1,75 @@
+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))
+
+# Define a dictionary of operations
+operations = {
+    ">": int.__gt__,
+    "<": int.__lt__
+}
+
+# Define a function to check if an item is accepted by a workflow
+def is_item_accepted(item, workflow_name = "in"):
+    if workflow_name == "R":
+        return False
+    if workflow_name == "A":
+        return True
+
+    rules, fallback = workflows_dict[workflow_name]
+    
+    for comparison_key, comparison_operator, comparison_value, target in rules:
+        if operations[comparison_operator](item[comparison_key], comparison_value):
+            return is_item_accepted(item, target)
+    
+    return is_item_accepted(item, fallback)
+
+# Initialize a variable to store the total
+total = 0
+
+# Process the second block of input
+for line in block2.splitlines():
+    item = {}
+    for segment in line[1:-1].split(","):
+        character, number = segment.split("=")
+        item[character] = int(number)
+    if is_item_accepted(item):
+        total += sum(item.values())
+
+# Print the total
+print(total)

2023/19/part 2 solution.py

@@ -0,0 +1,82 @@
+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"}))