117 lines
2.4 KiB
Python
117 lines
2.4 KiB
Python
import math
|
|
INPUT_FILE = "input.text"
|
|
|
|
I = 1000000
|
|
if I != 1:
|
|
I -= 1
|
|
|
|
class Point:
|
|
x: int
|
|
y: int
|
|
|
|
def __init__(self, x: int , y: int):
|
|
self.x = x
|
|
self.y = y
|
|
|
|
def add(self, other):
|
|
return Point(self.x + other.x, self.y + other.y)
|
|
|
|
def __add__(self, other):
|
|
return Point(self.x + other.x, self.y + other.y)
|
|
|
|
def __sub__(self, other):
|
|
return Point(self.x - other.x, self.y - other.x)
|
|
|
|
def __str__(self) -> str:
|
|
return f'P({self.x}, {self.y})'
|
|
|
|
def __eq__(self, other) -> bool:
|
|
return self.x == other.x and self.y == other.y
|
|
|
|
def _length(self) -> float:
|
|
return math.sqrt(self.x ** 2 + self.y ** 2)
|
|
|
|
def normalise(self) -> None:
|
|
self.x /= self._length()
|
|
self.y /= self._length()
|
|
|
|
def _abs(self) -> None:
|
|
self.x = abs(self.x)
|
|
self.y = abs(self.y)
|
|
|
|
def main():
|
|
|
|
with open(INPUT_FILE, 'r') as file:
|
|
universe = file.readlines()
|
|
|
|
# Shave off the newlines
|
|
for i in range(0, len(universe)):
|
|
universe[i] = universe[i][:-1]
|
|
|
|
# oh but first find galaxies
|
|
galaxies: list[Point] = []
|
|
n_y = len(universe)
|
|
|
|
for y in range(0, n_y):
|
|
row = universe[y]
|
|
|
|
for x in range(0, len(row)):
|
|
if row[x] == '#':
|
|
galaxies.append(Point(x, y))
|
|
|
|
# Check for expansion (y direction)
|
|
expansions = 0
|
|
for y in range(0, len(universe)):
|
|
row: str = universe[y]
|
|
|
|
if row.find('#') == -1:
|
|
for g in galaxies:
|
|
if g.y > y + (expansions * I):
|
|
g.y += I
|
|
expansions += 1
|
|
|
|
# Check for expansions (x direction)
|
|
expansions = 0
|
|
for x in range(0, len(universe[0])):
|
|
|
|
no_galaxy = True
|
|
|
|
for y in range(0, len(universe)):
|
|
if universe[y][x] == '#':
|
|
no_galaxy = False
|
|
break
|
|
|
|
if no_galaxy:
|
|
for g in galaxies:
|
|
if g.x > x + (expansions * I):
|
|
g.x += I
|
|
expansions += 1
|
|
|
|
# ok now find lengths or something
|
|
total = 0
|
|
for g1 in galaxies:
|
|
for g2 in galaxies:
|
|
|
|
if g2.y > g1.y:
|
|
dy = g2.y - g1.y
|
|
else:
|
|
dy = g1.y - g2.y
|
|
|
|
if g2.x > g1.x:
|
|
dx = g2.x - g1.x
|
|
else:
|
|
dx = g1.x - g2.x
|
|
|
|
total += dx + dy
|
|
|
|
|
|
|
|
print(f'Sum is {int(total/2)}')
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|