sm/maze
1
0
Fork 0
Code Issues Pull Requests Activity

fix to keep track of path

Browse Source
This commit is contained in:
Stephen McQuay 2012-10-31 22:03:37 -07:00
parent c0be71036d
commit 091232a21c
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
maze/__init__.py
View File

@ -12,7 +12,7 @@ class CS235Maze(object):
self._data = data
self._size = len(data)
def _solve(self, x, y, z):
def _solve(self, x, y, z, path=None):
if x < 0 or y < 0 or z < 0 or \
x >= self._size or y >= self._size or z >= self._size:
return False
@ -21,22 +21,27 @@ class CS235Maze(object):
elif x == self._size - 1 and y == self._size - 1 \
and z == self._size - 1:
self._data[x][y][z] = PATH
path.append((x, y, z))
return True
else:
self._data[x][y][z] = PATH
if self._solve(x - 1, y, z) or \
self._solve(x + 1, y, z) or \
self._solve(x, y - 1, z) or \
self._solve(x, y + 1, z) or \
self._solve(x, y, z - 1) or \
self._solve(x, y, z + 1):
if self._solve(x - 1, y, z, path) or \
self._solve(x + 1, y, z, path) or \
self._solve(x, y - 1, z, path) or \
self._solve(x, y + 1, z, path) or \
self._solve(x, y, z - 1, path) or \
self._solve(x, y, z + 1, path):
path.append((x, y, z))
return True
else:
self._data[x][y][z] = DEAD
return False
def solve(self):
return self._solve(0, 0, 0)
path = []
success = self._solve(0, 0, 0, path)
path.reverse()
return success, path
def __str__(self):
r = derp_transpose(self._data)