题目描述:请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 a b c e s f c s a d e e 矩阵中包含一条字符串"bccced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
a b t g
c f c s
j d e h
解题思路:利用回溯法
//首先在矩阵中选一个格子作为路径的起点。假设矩阵中某个格子的字符为ch,并且这个格子将对应于路径上的第i个字符。
//如果路径上的第i个字符不是ch,那么这个格子不可能处在路径上的第i个位置。
//如果路径上的第i个字符正好是ch, 那么到相邻格子寻找路径上的第i+1个字符。除矩阵边界上的格子外,其他格子都有4个相邻的格子
由于回溯法的递归特性,路径可以看作一个栈。当在矩阵中定位了路径中前n个字符的位置之后,在与第n个字符对应的格子的周围都没有找到第n+1个字符,这时候只好在路径上回到第n-1个字符,重新定位第n个字符。
还需要一个bool矩阵来标识路径是否已经进入了每个格子。
class Solution{
public:
bool hasPath(char* matrix, int rows, int cols, char* str){
if(matrix == NULL || rows < 1 || cols < 1 || str == NULL)
return false;
bool *visited = new bool[rows*cols];
memset(visited, 0, rows*cols];
int pathLength = 0;
for(int row = 0; row < rows; ++row)
for(int col = 0; col < cols, ++col)
if(hasPathCore(matrix, rows, cols, row, col, str,
pathLength, visited))
return true;
delete[] visited;
return false;
}
bool hasPathCore(char* matrix, int rows, int cols, int row, int col,
const char* str, int& pathLength, bool* visited){
//如果路径上的所有字符都在矩阵中找到合适的位置,则此时str[pathLength] == '\0'
if(str[pathLength] == '\0')
return true;
bool hasPath = false;
if(row >= 0 && row < rows && col >= 0 && col <cols
&&matrix[row*cols + col] == str[pathLength]
&&!visited[row*cols + col]){
++pathLength;
visited[row*cols + col] = true;
hasPath = hasPathCore(matrix, rows, cols, row, col-1, str, pathLength, visited)
|| hasPathCore(matrix, rows, cols, row-1, col, str, pathLength, visited)
|| hasPathCore(matrix, rows, cols, row, col+1, str, pathLength, visited)
||hasPathCore(matrix, rows, cols, row+1, col, str, pathLength, visited);
if(!hasPath){//如果前进一步之后第n+1个字符不在合适位置,则向后退一步,并将bool矩阵改为false
--pathLength;
visited[row*cols + col] = false;
}
}
return hasPath;
}
};