day04.py

import csv
from functools import cached_property

class Assignments:
  @classmethod
  def has_containment(cls, range1, range2):
    set1 = set(range1)
    set2 = set(range2)
    intersection = set1.intersection(set2)
    return intersection == set1 or intersection == set2

  @classmethod
  def has_overlap(cls, range1, range2):
    set1 = set(range1)
    set2 = set(range2)
    intersection = set1.intersection(set2)
    return len(intersection) > 0

  def __init__(self, filename):
    self.pairs = []
    with open(filename, newline='') as csv_file:
      reader = csv.reader(csv_file)
      for row in reader:
        if len(row) > 0:
          self.pairs.append(self._parse_row(row))
  
  @cached_property
  def containment_count(self):
    return sum([1 if Assignments.has_containment(ranges[0], ranges[1]) else 0 for ranges in self.pairs])
  
  @cached_property
  def overlap_count(self):
    return sum([1 if Assignments.has_overlap(ranges[0], ranges[1]) else 0 for ranges in self.pairs])

  def _parse_row(self, row):
    return [self._parse_range(row[0]), self._parse_range(row[1])]

  def _parse_range(self, range_string):
    range_elements = range_string.split('-')
    return range(int(range_elements[0]), int(range_elements[1]) + 1)


def main():
  assignments = Assignments('../inputs/2022/day04.csv')
  print(f'Part 1: {assignments.containment_count}')
  print(f'Part 2: {assignments.overlap_count}')

if __name__ == "__main__":
  main()