矩阵中的路径
题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如
矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
public class Solution {
public boolean hasPath(char[] matrix, int rows, int cols, char[] str)
{
//标志位,初始化位false
boolean flag[] = new boolean[matrix.length];
for(int i = 0; i < rows; i++) {
for (int j =0; j < cols; j++) {
//循环遍历二维数组,找到起点等于str第一个元素的值,再递归判断四周是否有符合条件的----回溯法
if (judge(matrix,i,j,rows,cols,flag,str,0)) {
return true;
}
}
}
return false;
}
////judge(初始矩阵,索引行坐标i,索引纵坐标j,矩阵行数,矩阵列数,待判断的字符串,字符串索引初始为0即先判断字符串的第一位)
private boolean judge(char[] matrix, int i, int j, int rows, int cols, boolean flag[], char[] str, int k) {
//先根据i和j计算匹配的第一个元素转为一维数组的位置
int index = i * cols + j;
//递归终止条件
if (i < 0 || j < 0 || i >= rows || j >= cols || matrix[index] != str[k] || flag[index] == true) {
return false;
}
//若k已经到达str末尾了,说明之前的都已经匹配成功了,直接返回true即可
if (k == str.length - 1) {
return true;
}
//第一次运行到这里时表示:要走的第一个位置置为true,表示已经走过了;递归时表示将当前位置加入到路径中,然后在if中判断当前位置的左右上下时候可以加入路径
flag[index] = true;
//回溯,递归寻找(当前元素的左边、右边、上边、下边元素,是否至少有一个可以构成下一个路径节点),每次找到了就给k加一,找不到,还原
if (judge(matrix,i,j - 1,rows,cols,flag,str,k + 1) ||
judge(matrix,i,j + 1,rows,cols,flag,str,k + 1) ||
judge(matrix,i - 1,j,rows,cols,flag,str,k + 1) ||
judge(matrix,i + 1,j,rows,cols,flag,str,k + 1) ) {
return true;
}
//当前节点的左右上下元素均不满足加入路径的条件,只能回溯到上一次入选节点位置,flag标志位还原为false
flag[index] = false;
return false;
}
}