03.py

"""--- Day 3: Squares With Three Sides ---"""

import re

with open("input/03.txt", "r") as f:
  valid_1 = valid_2 = 0

  a, b, c = [], [], []

  for instruction in f:
    sides = [int(x) for x in re.findall(" [0-9]+", instruction)]

    # this is for part 2
    a.append(sides[0])
    b.append(sides[1])
    c.append(sides[2])

    # solve part 1
    sides.sort()
    if sides[0] + sides[1] > sides[2]:
      valid_1 += 1

  print(valid_1)

  for i in range(0, len(a), 3):
    # row a
    sides = a[i:i+3]
    sides.sort()
    if sides[0] + sides[1] > sides[2]:
      valid_2 += 1

    # row b
    sides = b[i:i+3]
    sides.sort()
    if sides[0] + sides[1] > sides[2]:
      valid_2 += 1

    # row c
    sides = c[i:i+3]
    sides.sort()
    if sides[0] + sides[1] > sides[2]:
      valid_2 += 1

  print(valid_2)