请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 a b c e s f c s a d e e 矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子
思路:因为矩阵中的每个点都可能是起点,所以需要遍历。然后我们就可以使用典型的回溯剪枝的方法进行下一步工作了,也就是我们熟悉的递归。同时因为不允许重复进入同一个格子,所以需要使用boolean数组进行控制。
public class Solution {
boolean [] visible= null; // 记录并避免重复进入同一个格子的情况
public boolean hasPath(char[] matrix, int rows, int cols, char[] str) {
visible = new boolean[matrix.length];
for(int i = 0;i<rows;i++){ // 矩阵中每个点都可能是起点,开启遍历
for(int j = 0;j<cols;j++){
if(hasSubPath(matrix,rows,cols,str,i,j,0)){ // 最后一位0为记录从字符串的首个字符开始
return true;
}
}
}
return false;
}
//采用递归(回溯剪枝),我们只关心本步该怎么走。
// rows总行数,cols总列数,col当前列数,row当前行数,len当前串长度
// row*cols+col为当前数字在矩阵中的索引
public boolean hasSubPath(char[] matrix, int rows, int cols, char[] str,int row,int col,int len){
if(str[len]!=matrix[row*cols+col]||visible[row*cols+col])return false; // 当前值与矩阵当前索引处值不相等或者当位置已经被标记(避免重复)
if(str.length-1==len)return true; // 长度已经匹配,子串匹配成功
visible[row*cols+col] = true; //未验证当前字符前先标记为true
if(row>0&&hasSubPath(matrix,rows,cols,str,row-1,col,len+1))return true; // 上一个
if(col>0&&hasSubPath( matrix,rows,cols,str,row,col-1,len+1))return true; // 左一个
if(row<rows-1&&hasSubPath(matrix,rows,cols,str,row+1,col,len+1))return true; // 下一个
if(col<cols-1&&hasSubPath(matrix,rows,cols,str,row,col+1,len+1))return true; // 右一个
visible[row*cols+col] = false; // 验证完该字符标记为false,因题意不能重复进入同一格子
return false;
}
}
- 时间复杂度:O(mn)。
- 空间复杂度:O(n)。
练习地址:https://www.nowcoder.com/practice/c61c6999eecb4b8f88a98f66b273a3cc