题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如
矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
思路:回溯
每走一格就判断是否和字符相符,如果相符就判断是否在上下左右也相符,如果不相符就回退一格
class Solution {
public:
bool hasPath(char* matrix, int rows, int cols, char* str)
{
if(str == NULL) return false;
bool *visited = new bool[rows*cols];
for(int i = 0;i < rows*cols;i++)
visited[i] = false;
for(int i = 0;i < rows;i++){
for(int j = 0;j < cols;j++){
if(dfs(matrix,rows,cols,str,i,j,visited)){
delete [] visited;//别忘了释放空间
return true;
}
}
}
delete [] visited;//别忘了释放空间
return false;
}
bool dfs(char* matrix, int rows, int cols, char* str,int i,int j,bool* visited)
{
if((*str) == '\0')//字符串走到头了
return true;
//判断(i,j)出界或已被访问过
if(i >= rows || j >= cols || i < 0 || j < 0 || visited[i*cols+j])
return false;
if(matrix[i * cols + j] != (*str))
return false;
visited[i * cols + j] = true;
bool ok = dfs(matrix,rows,cols,str+1,i-1,j,visited)||
dfs(matrix,rows,cols,str+1,i+1,j,visited)||
dfs(matrix,rows,cols,str+1,i,j-1,visited)||
dfs(matrix,rows,cols,str+1,i,j+1,visited);
if(ok == false)
visited[i * cols + j] = false;//这一格走不通,回退到上一格
return ok;
}
};