import 2021

2021/day8/day8.py

@@ -0,0 +1,46 @@
+def part1(inp):
+    total = 0
+    for display in inp:
+        _, output_values = display
+        for value in output_values:
+            if len(value) in [2, 4, 3, 7]:
+                total += 1
+    print(f"Part 1 : {total}")
+
+
+def part2(inp):
+    for display in inp:
+        patterns, values = display
+        patterns = sorted(patterns, key=lambda x: len(x))
+        print(patterns)
+        # easy
+        d1 = [x for x in patterns if len(x) == 2][0]
+        print("1", d1)
+        d4 = [x for x in patterns if len(x) == 4][0]
+        print("4", d4)
+        d7 = [x for x in patterns if len(x) == 3][0]
+        print("7", d7)
+        d8 = [x for x in patterns if len(x) == 7][0]
+        print("8", d8)
+
+        # 3 is the only digit that has all common segments with 1
+        breakpoint()
+        d3 = [x for x in patterns if set(d1).issubset(set(x)) and len(x) == 5][0]
+        print("3", d3)
+
+        break
+
+def main(infile):
+    inp = []
+    with open(infile) as f:
+        for display in f:
+            display = display.rstrip().split(" | ")
+            signal_patterns = display[0].split(" ")
+            output_values = display[1].split(" ")
+            inp.append([signal_patterns, output_values])
+    part1(inp)
+    part2(inp)
+
+
+if __name__ == "__main__":
+    main("input.txt")

2021/day8/input.txt

@@ -0,0 +1 @@
+acedgfb cdfbe gcdfa fbcad dab cefabd cdfgeb eafb cagedb ab | cdfeb fcadb cdfeb cdbaf