did day6 p1 and started p2

2024/day6/main.py

@@ -0,0 +1,96 @@
+with open('input.text', 'r') as file:
+    data: list[str] = file.readlines()
+
+y_bound = len(data)
+x_bound = len(data[0])
+
+class Vector:
+    x: int
+    y: int
+
+    def __init__(self, x: int, y: int):
+        self.x = x
+        self.y = y
+
+    def __add__(self, other):
+        return Vector(self.x + other.x, self.y + other.y)
+
+    def __str__(self):
+        return f'{self.x}, {self.y}'
+    
+    def __eq__(self, value):
+        return self.x == value.x and self.y == value.y
+
+class Guard:
+    pos: Vector
+    dir: Vector
+
+    def __init__(self, p: Vector, d: Vector):
+        self.pos = p
+        self.dir = d
+
+    def move(self, ob) -> bool:
+        new_pos = self.pos + self.dir
+
+        # Check if need to turn
+        if new_pos in ob:
+            self.turn()
+            return self.move(ob)
+
+        # Check if move out of bounds
+        if new_pos.y < 0 or new_pos.y >= y_bound or new_pos.x < 0 or new_pos.x >= x_bound:
+            return False
+        
+        self.pos = new_pos
+        return True
+
+
+    def turn(self):
+        temp_x = self.dir.x
+        self.dir.x = -self.dir.y
+        self.dir.y = temp_x
+
+# Parse Data
+obstacles = []
+guard = None
+
+for y, line in enumerate(data):
+    for x, char in enumerate(line):
+        match char:
+            case '#':
+                obstacles.append(Vector(x, y))
+            case '^':
+                guard = Guard(Vector(x, y), Vector(0, -1))
+
+starting_pos = guard.pos
+
+# Part 1
+been = [starting_pos]
+
+while guard.move(obstacles):
+    if guard.pos not in been:
+        been.append(guard.pos)
+
+print(f'Part 1: {len(been)}')
+
+# Part 2
+total = 0
+
+been.remove(starting_pos)
+for place in been:
+    temp_obstacles = obstacles.copy()
+    temp_obstacles.append(place)
+
+    guard.pos = starting_pos
+    guard.dir = Vector(0, -1)
+
+    steps = 0
+    while guard.move(temp_obstacles):
+        steps += 1
+
+        # Yup, this is what I count as in a loop...
+        if steps >= 10 ** 5:
+            total += 1
+            break
+
+print(f'Part 2: {total}')