[py] Port 2016_03

py/aoc/y2016/d03.py

@@ -0,0 +1,45 @@
+import re
+
+TRIANGLE_RE = r"\s*(\d+)\s+(\d+)\s+(\d+)"
+
+
+def load_triangles(inputstr):
+    triangles = []
+    for line in inputstr.splitlines():
+        match = re.fullmatch(TRIANGLE_RE, line)
+        a, b, c = match.groups()
+        a, b, c = int(a), int(b), int(c)
+        triangles.append((a, b, c))
+    return triangles
+
+
+# PART 1
+
+
+def is_valid(triangle):
+    a, b, c = sorted(triangle)
+    return a + b > c
+
+
+def count_valid(triangles):
+    return len(list(filter(is_valid, triangles)))
+
+
+# PART 2
+
+
+def transform_triangles(triangles):
+    new_triangles = []
+    while triangles:
+        t1, t2, t3 = triangles[:3]
+        triangles = triangles[3:]
+        new_triangles.extend(zip(t1, t2, t3))
+    return new_triangles
+
+
+def solve(inputstr):
+    triangles = load_triangles(inputstr)
+    valid = count_valid(triangles)
+    print(f"Part 1: {valid}")
+    valid_2 = count_valid(transform_triangles(triangles))
+    print(f"Part 2: {valid_2}")