You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

49 lines
1.1 KiB

3 years ago
def readLines():
with open("input") as file:
lines = []
for line in file.read().splitlines():
p1, p2 = line.split(" -> ")
x1, y1 = p1.split(",")
x2, y2 = p2.split(",")
lines.append((int(x1), int(y1), int(x2), int(y2)))
return lines
def filterDiagonals(lines):
new = []
for x1, y1, x2, y2 in lines:
if x1 == x2 or y1 == y2:
new.append((x1, y1, x2, y2))
return new
def expandToPointList(line):
x1, y1, x2, y2 = line
xDir, yDir = x2 - x1, y2 - y1
count = max(abs(xDir), abs(yDir))
xDir = xDir // abs(xDir) if xDir != 0 else 0
yDir = yDir // abs(yDir) if yDir != 0 else 0
points = []
for i in range(count + 1):
x, y = x1 + xDir * i, y1 + yDir * i
points.append((x, y))
return points
def getMostDangerousCount(lines):
points, intersections = set(), set()
for line in lines:
linePoints = expandToPointList(line)
intersections = intersections.union(points.intersection(linePoints))
points = points.union(linePoints)
return len(intersections)
def solve():
allLines = readLines()
print(getMostDangerousCount(filterDiagonals(allLines)))
print(getMostDangerousCount(allLines))
solve()