题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则之后不能再次进入这个格子。
例如 a b c e s f c s a d e e 这样的3 X 4 矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
解法一:
遍历矩阵,找到和 str 中第一个字符相等的字符,标记该字符在矩阵的位置,防止重复出现
分别在矩阵中的左右上下四个方向,查找与 str 中下一个字符相等的字符,接着标记位置
重复第二步,直到 str 的最后一个字符 ‘\0’;或者找不到匹配为止
如果可以到 str 的最后一个字符,则返回 true
如果在某一位置不能继续匹配,则返回第一步继续执行,直到遍历完矩阵为止
如果遍历完矩阵,则返回 false
NOTE:
因为是采用一维数组存储的矩阵数据,所以在遍历和向上下左右四个方向走的时候,要注意一下。
class Solution {
public:
bool hasPath(char* matrix, int rows, int cols, char* str)
{
if (*str == '\0')
return true;
vector<int> vec(rows * cols, 0);
for (int i = 0; i < rows; i++)
for (int j = 0; j < cols; j++)
{
if (matrix[i * cols + j] == *str && findPath(matrix, rows, cols, str, vec, i, j))
return true;
}
return false;
}
private:
bool findPath(char *matrix, int rows, int cols, char *str, vector<int> &vec, int x, int y)
{
//这句话的位置一定要在前边
//如果str到末尾了,就证明此路径是存在的
if (*str == '\0')
return true;
//保证在矩阵范围内
if (x < 0 || x >= rows || y < 0 || y >= cols)
return false;
if (vec[x * cols + y] == 1 || matrix[x * cols + y] != *str)
return false;
// 如果还没到末尾
vec[x * cols + y] = 1;
bool sign = findPath(matrix, rows, cols, str + 1, vec, x + 1, y) || findPath(matrix, rows, cols, str + 1, vec, x - 1, y) || findPath(matrix, rows, cols, str + 1, vec, x, y + 1) || findPath(matrix, rows, cols, str + 1, vec, x, y - 1);
vec[x * cols + y] = 0;
return sign;
}
};