题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 a b c e s f c s a d e e 矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
思路:典型的BFS,先寻找等于第一个路径字符的位置,从该位置,像四个方向遍历,当所有路径字符均被把遍历后,则找的到该路径。
public class Solution {
private final int dx[] = {1, -1, 0, 0};
private final int dy[] = {0, 0, 1, -1};
public boolean jude(char[] matrix, int rows, int cols, char[] str, int k, int index, boolean[] vis) {
//k表示当前匹配长度,index表示当前匹配矩阵中的位置
vis[index] = true;
if (k == str.length) { // 匹配成功
return true;
}
int x = index / cols;
int y = index % cols;
for (int i = 0; i < 4; i++) { // 向四个方向走
int X = x + dx[i];
int Y = y + dy[i];
int next = X * cols + Y;
if (X >= 0 && Y >= 0 && X < rows && Y < cols && !vis[next] && matrix[next] == str[k] && jude(matrix, rows, cols, str, k + 1, next, vis)) {
return true;
}
}
vis[index] = false; // 处理完index子问题
return false;
}
public boolean hasPath(char[] matrix, int rows, int cols, char[] str) {
if (matrix.length==0) {
return false;
}
boolean[] vis = new boolean[matrix.length]; // 标记数组
for (int i = 0; i < matrix.length; i++) {
if (matrix[i] == str[0] && jude(matrix, rows, cols, str, 1, i, vis)) {
return true;
}
}
return false;
}
}