[2] добавил поиск в ширину

This commit is contained in:
raskatovia 2026-05-22 21:26:10 +03:00
parent a912487cdc
commit 308ee2d716

View File

@ -0,0 +1,58 @@
from collections import deque
from maze import MazeBuilder
def build_path(previous, start, finish):
if finish not in previous:
return []
path = []
current = finish
while current != start:
path.append(current)
current = previous[current]
path.append(start)
path.reverse()
return path
class BfsStrategy:
def solve(self, maze):
start = maze.start
finish = maze.finish
queue = deque([start])
previous = {start: None}
visited_count = 0
while queue:
current = queue.popleft()
visited_count += 1
if current == finish:
break
for next_cell in maze.neighbors(current[0], current[1]):
if next_cell not in previous:
previous[next_cell] = current
queue.append(next_cell)
path = build_path(previous, start, finish)
return {
"name": "BFS",
"path": path,
"visited": visited_count,
"length": len(path)
}
class MazeSolver:
def __init__(self, strategy):
self.strategy = strategy
def solve(self, maze):
return self.strategy.solve(maze)
if __name__ == "__main__":
maze = MazeBuilder().from_file("raskatovia/docs/data/task2/maps/simple.txt").build()
solver = MazeSolver(BfsStrategy())
result = solver.solve(maze)
print("algorithm:", result["name"])
print("visited:", result["visited"])
print("length:", result["length"])
print(maze.draw(result["path"]))