This commit is contained in:
Dmitry Fedotov
2025-03-22 18:38:32 +03:00
parent b0a982c64d
commit 88def4674b
3 changed files with 22 additions and 9 deletions

1
.gitignore vendored Normal file
View File

@@ -0,0 +1 @@
lib/__pycache__

View File

@@ -50,7 +50,7 @@ class Tile(object):
self.links = [S, E] self.links = [S, E]
def __str__(self): def __str__(self):
return f'{self.val} x: {self.x} y: {self.y}' return f'{self.val}' # x: {self.x} y: {self.y}'
def __eq__(self, obj) -> bool: def __eq__(self, obj) -> bool:
return self.x == obj.x and self.y == obj.y return self.x == obj.x and self.y == obj.y
@@ -126,10 +126,12 @@ class Walker(Field):
while True: while True:
self.loop.append(curr) self.loop.append(curr)
nxt = self.get_adjacent(curr, dir) nxt = self.get_adjacent(curr, dir)
if not nxt or not nxt.has_link_from(dir.opposite()): if not nxt:
break break
if nxt.val == START: if nxt.val == START:
return return
if not nxt.has_link_from(dir.opposite()):
break
curr = nxt curr = nxt
dir = nxt.opposite_pipe_end(dir.opposite()) dir = nxt.opposite_pipe_end(dir.opposite())
@@ -147,17 +149,27 @@ class Walker(Field):
inside_loop = set() inside_loop = set()
for t in self.loop: for t in self.loop:
for d in [N, E, S, W]: for d in [N, E, S, W]:
newt = self.get_adjacent(t, d) newt = self.get_adjacent(t, d)
if not newt or newt in self.visited: if not newt or newt in self.visited:
continue continue
if self._is_inside_loop(newt): if self._is_inside_loop(newt, [N, E, S, W]):
inside_loop.add(newt) inside_loop.add(newt)
self.visited.add(newt) self.visited.add(newt)
return len(inside_loop) return len(inside_loop)
def _is_inside_loop(self, t: Tile) -> bool: def _is_inside_loop(self, t: Tile, dirs: list[Direction]) -> bool:
# TODO if not t:
return False
if t in self.loop:
return True
for d in dirs:
newt = self.get_adjacent(t, d)
if not self._is_inside_loop(newt, [d]):
return False
return True return True
@@ -167,8 +179,7 @@ if __name__ == '__main__':
lines = r.read() lines = r.read()
w = Walker(lines) w = Walker(lines)
steps = w.find_max_steps_in_loop() print(w)
print(steps)
n = w.find_captured_by_loop() n = w.find_captured_by_loop()
print(n) print(n)
@@ -178,5 +189,6 @@ if __name__ == '__main__':
w = Walker(lines) w = Walker(lines)
steps = w.find_max_steps_in_loop() steps = w.find_max_steps_in_loop()
print(steps) print(steps)
# print(w.loop) n = w.find_captured_by_loop()
print(n)