init

2023/day11/p2/main.py

@@ -0,0 +1,116 @@
+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()