41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
from tools.aoc import AOCDay
|
|
from tools.coordinate import Coordinate
|
|
from tools.grid import Grid
|
|
from typing import Any, List
|
|
|
|
|
|
def buildLineGrid(lines: List[List[str]], diagonals: bool = False) -> Grid:
|
|
grid = Grid(0)
|
|
for line in lines:
|
|
start = Coordinate(*map(int, line[0].split(",")))
|
|
end = Coordinate(*map(int, line[1].split(",")))
|
|
if not diagonals and start.x != end.x and start.y != end.y:
|
|
continue
|
|
|
|
for c in start.getLineTo(end):
|
|
grid.add(c)
|
|
|
|
return grid
|
|
|
|
|
|
def getCrossCount(grid: Grid) -> int:
|
|
count = 0
|
|
for x in grid.getActiveCells():
|
|
count += grid.get(x) > 1
|
|
|
|
return count
|
|
|
|
|
|
class Day(AOCDay):
|
|
test_solutions_p1 = [5, 6710]
|
|
test_solutions_p2 = [12, 20121]
|
|
|
|
def part1(self) -> Any:
|
|
grid = buildLineGrid(self.getInputAsArraySplit(" -> "))
|
|
return getCrossCount(grid)
|
|
|
|
def part2(self) -> Any:
|
|
grid = buildLineGrid(self.getInputAsArraySplit(" -> "), True)
|
|
return getCrossCount(grid)
|
|
|