Add day07

Author: Mishco

2015/day-07/main.py

@@ -0,0 +1,103 @@
+# --- Day 7: Some Assembly Required ---
+from collections import defaultdict
+import operator
+
+def part1():
+    # read input
+    # all possible gates: AND, OR, LSHIFT, RSHIFT, NOT
+    ops = {'AND': operator.and_, 'OR': operator.or_, 'LSHIFT': operator.lshift, 'RSHIFT': operator.rshift}
+    max_val = 65535
+
+    lines = open("../inputs/day-07.txt").readlines()
+    output = {}
+    remaining = list(lines)
+    while remaining:
+        unresolved = []
+        for line in remaining:
+            items = line.strip().split('->')
+            granular_items = items[0].split()
+            res = items[1].strip()
+            if len(granular_items) == 1:
+                token = items[0].strip()
+                if token.isdigit():
+                    output[res] = int(token)
+                elif token in output:
+                    output[res] = output[token]
+                else:
+                    unresolved.append(line)
+                    continue
+            elif len(granular_items) > 2:
+                first_gran_item = granular_items[0]
+                second_gran_item = granular_items[1]
+                third_gran_item = granular_items[2]
+                left_val = int(first_gran_item) if first_gran_item.isdigit() else output.get(first_gran_item)
+                right_val = int(third_gran_item) if third_gran_item.isdigit() else output.get(third_gran_item)
+                if left_val is None or right_val is None:
+                    unresolved.append(line)
+                    continue
+                output[res] = ops[second_gran_item](left_val, right_val)
+            else:
+                second_gran_item = granular_items[1]
+                if second_gran_item not in output:
+                    unresolved.append(line)
+                    continue
+                output[res] = max_val - output[second_gran_item]
+            output[res] &= max_val
+        if len(unresolved) == len(remaining):
+            break # to avoid infinite loop
+        remaining = unresolved
+
+    return output['a']
+
+
+def part2():
+    ops = {'AND': operator.and_, 'OR': operator.or_, 'LSHIFT': operator.lshift, 'RSHIFT': operator.rshift}
+    max_val = 65535
+
+    lines = open("../inputs/day-07.txt").readlines()
+    a_signal = part1()
+
+    output = {'b': a_signal}
+    remaining = [line for line in lines if line.strip().split('->')[1].strip() != 'b']
+    while remaining:
+        unresolved = []
+        for line in remaining:
+            items = line.strip().split('->')
+            granular_items = items[0].split()
+            res = items[1].strip()
+            if len(granular_items) == 1:
+                token = items[0].strip()
+                if token.isdigit():
+                    output[res] = int(token)
+                elif token in output:
+                    output[res] = output[token]
+                else:
+                    unresolved.append(line)
+                    continue
+            elif len(granular_items) > 2:
+                first_gran_item = granular_items[0]
+                second_gran_item = granular_items[1]
+                third_gran_item = granular_items[2]
+                left_val = int(first_gran_item) if first_gran_item.isdigit() else output.get(first_gran_item)
+                right_val = int(third_gran_item) if third_gran_item.isdigit() else output.get(third_gran_item)
+                if left_val is None or right_val is None:
+                    unresolved.append(line)
+                    continue
+                output[res] = ops[second_gran_item](left_val, right_val)
+            else:
+                second_gran_item = granular_items[1]
+                if second_gran_item not in output:
+                    unresolved.append(line)
+                    continue
+                output[res] = max_val - output[second_gran_item]
+            output[res] &= max_val
+        if len(unresolved) == len(remaining):
+            break
+        remaining = unresolved
+
+    return output['a']
+
+
+if __name__ == "__main__":
+    print("Part 1:", part1())
+    print("Part 2:", part2())