题目描述:
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则之后不能再次进入这个格子。
例如 a b c e s f c s a d e e 这样的3 X 4
矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
思路分析:
回溯法解决,从矩阵中一个点开始,如果该点的值等于字符串中第一个值,那么接下来我们可以从该点的左上右下节点继续搜索。但是不能重复遍历,如果最终能找到一条字符串中的路径那么返回true,如果不能返回false。
代码:
public class Solution {
public boolean hasPath(char []matrix,int rows,int cols,char[]str){
if(matrix==null||str==null)
return false;
boolean []flag=new boolean[matrix.length]; //标记是否被访问过
for(int i=0;i<rows;i++){
for(int j=0;j<cols;j++){
if(findPath(matrix,rows,cols,i,j,str,0,flag))
return true;
}
}
return false;
}
public boolean findPath(char []matrix,int rows,int cols,int i,int j,char[]str,int k,boolean[]flag){
int index=i*cols+j;
if(i<0||i>=rows||j<0||j>=cols||flag[index]||matrix[index]!=str[k])
return false;
if(k==str.length-1)
return true; //证明找到了该条路径
flag[index]=true; //当前点被访问过了
if(findPath(matrix,rows,cols,i,j-1,str,k+1,flag) //向左搜
||findPath(matrix,rows,cols,i-1,j,str,k+1,flag)//向上搜
||findPath(matrix,rows,cols,i,j+1,str,k+1,flag)//向右搜
||findPath(matrix,rows,cols,i+1,j,str,k+1,flag)//向下搜
)
return true;
flag[index]=false; //回溯
return false;
}
}