Day 03: Solution of part 2

Author: unstko

03/log.txt

@@ -12,8 +12,6 @@ In your puzzle input, how many of the listed triangles are possible?
 
 Your puzzle answer was 1050.
 
-The first half of this puzzle is complete! It provides one gold star: *
-
 --- Part Two ---
 
 Now that you've helpfully marked up their design documents, it occurs to you that triangles are specified in groups of three vertically. Each set of three numbers in a column specifies a triangle. Rows are unrelated.
@@ -26,4 +24,8 @@ For example, given the following specification, numbers with the same hundreds d
 201 401 601
 202 402 602
 203 403 603
-In your puzzle input, and instead reading by columns, how many of the listed triangles are possible?
+In your puzzle input, and instead reading by columns, how many of the listed triangles are possible?
+
+Your puzzle answer was 1921.
+
+Both parts of this puzzle are complete! They provide two gold stars: **

03/solution.py

@@ -8,6 +8,8 @@ def __init__(self, nr):
         super().__init__(nr)
         self.triangle_1 = Triangle()
         self.possible_1 = 0
+        self.triangles_2 = [Triangle(), Triangle(), Triangle()]
+        self.possible_2 = 0
 
     def calculate(self, test=False):
         self.read_instructions()
@@ -17,9 +19,18 @@ def read_instructions(self):
         self.read_input(True)
 
     def check_triangles(self):
+        group_part = 0
         for triangle in self.input:
             sides = re.split(r'\s+', triangle.strip(' '))
             self.triangle_1.set_sides(sides)
             if self.triangle_1.is_possible():
                 self.possible_1 += 1
+            for nr in range(0, 3):
+                self.triangles_2[nr].set_side(group_part, sides[nr])
+            if group_part == 2:
+                for nr in range(0, 3):
+                    if self.triangles_2[nr].is_possible():
+                        self.possible_2 += 1
+            group_part = (group_part + 1) % 3
         self.set_solution(1, self.possible_1)
+        self.set_solution(2, self.possible_2)