diff --git a/backtracking/rat_in_maze.py b/backtracking/rat_in_maze.py index 7bde886dd558..13b741706a5a 100644 --- a/backtracking/rat_in_maze.py +++ b/backtracking/rat_in_maze.py @@ -64,9 +64,9 @@ def solve_maze(maze: list[list[int]]) -> bool: solutions = [[0 for _ in range(size)] for _ in range(size)] solved = run_maze(maze, 0, 0, solutions) if solved: - print("\n".join(str(row) for row in solutions)) + solved = (solutions, 1) else: - print("No solution exists!") + solved = "No solution exists!" return solved @@ -113,6 +113,14 @@ def run_maze(maze: list[list[int]], i: int, j: int, solutions: list[list[int]]) if __name__ == "__main__": + maze = [ + [0, 1, 0, 1, 1], + [0, 0, 0, 0, 0], + [1, 0, 1, 0, 1], + [0, 0, 1, 0, 0], + [1, 0, 0, 1, 0], + ] + print(solve_maze(maze)) import doctest doctest.testmod()