题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则之后不能再次进入这个格子。 例如 a b c e s f c s a d e e 这样的3 X 4 矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
这一道题用回溯的思想去做
先贴一下代码
public class Solution {
public boolean hasPath(char[] matrix, int rows, int cols, char[] str){
boolean [] visitFlags = new boolean [matrix.length];
for(int row = 0; row < rows ; row++){
for(int col = 0; col < cols; col++){
if(helper(matrix , rows , cols , row , col , 0 , str , visitFlags)){
return true;
}
}
}
return false;
}
public boolean helper(char[] matrix , int rows , int cols , int row , int col , int k , char[] str , boolean [] visitFlags){
int index = row * cols + col;
if(row < 0 || col < 0 || row >= rows || col >= cols || visitFlags[index] || matrix[index] != str[k]){
return false;
}
visitFlags[index] = true;
if(str.length - 1 == k){
return true;
}
k++;
if(helper(matrix , rows , cols , row - 1 , col , k , str , visitFlags)
|| helper(matrix , rows , cols , row + 1 , col , k , str , visitFlags)
|| helper(matrix , rows , cols , row , col -1 , k , str , visitFlags)
|| helper(matrix , rows , cols , row , col + 1 , k , str , visitFlags)){
return true;
}
visitFlags[index] = false;
return false;
}
}
因为matrix是一维的,所以按照index查询的时候,注意index的取值
main函数遍历每一个位置,然后helper函数的作用是查看当前查到的字符是否满足要求。