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}")