Day14.py

from helpers.importHelpers import *

def insertLine(line, caveMap):
  coords = line.split(" -> ")
  for i in range(len(coords)-1):
    x1, y1 = coords[i].split(",")
    x2, y2 = coords[i+1].split(",")
    x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
    stepX = 1 if x2>=x1 else -1
    stepY = 1 if y2>=y1 else -1
    for x in range(x1, x2+stepX, stepX):
      for y in range(y1, y2+stepY, stepY):
        caveMap.add((x,y))

def getCaveMap(stringInput):
  caveMap = set()
  for line in stringInput.split("\n"):
    insertLine(line, caveMap)
  return caveMap

def letTheSandFall(caveMap, source = 500):
  part1Done = False
  part1, part2 = 0, 0
  maxDepth = max(caveMap, key=lambda x: x[1])[1]
  floor = maxDepth + 2
  x = source
  y = 0
  while (source, 0) not in caveMap:
    #look ahead where to go next
    if (x,y+1) in caveMap or y+1 == floor: #if there is a floor or sand below
      if (x-1,y+1) not in caveMap and y+1 != floor: #if there is no floor or sand to the left, go left
        x -= 1
      elif (x+1,y+1) not in caveMap and y+1 != floor: #if there is no floor or sand to the right, go right
        x += 1
      else: #else, stay here
        caveMap.add((x,y))
        if not part1Done and y >= maxDepth:
          part1 = part2
          part1Done = True
        part2 += 1
        x = source
        y = -1
    y += 1
  return part1, part2

part1, part2 = letTheSandFall(getCaveMap(getInput()))
print("Part 1: ", part1)
print("Part 2: ", part2)