题目
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。
路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。
如果一条路径经过了矩阵中的某一个格子,则之后不能再次进入这个格子。
注意:
输入的路径不为空;
所有出现的字符均为大写英文字母;
样例
matrix=
[
[“A”,”B”,”C”,”E”],
[“S”,”F”,”C”,”S”],
[“A”,”D”,”E”,”E”]
]
str=”BCCE” , return “true”
str=”ASAE” , return “false”
解法:dfs搜索
首先枚举字符串的起点,然后依次搜索其邻居
由于一条路径中不能重复访问同一个位置,因此我们需要在访问完之后修改掉该位置元素的值,此题不适合用额外数组记录的方式
时间复杂度O(n^2 3^k),空间复杂度O(1)
class Solution {
public:
bool hasPath(vector<vector<char>>& matrix, string &str) {
for (int i = 0; i < matrix.size(); i++)
for (int j = 0; j < matrix[i].size(); j++)
if (dfs(matrix, i, j, 0, str))
return true;
return false;
}
bool dfs(vector<vector<char>> &matrix, int x, int y, int p, const string &str) {
if (matrix[x][y] != str[p]) return false;
if (p >= str.size() - 1) return true;
char tmp = matrix[x][y]; // 替换
matrix[x][y] = '#';
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
for (int i = 0; i < 4; i++) {
int nx = x + dx[i], ny = y + dy[i];
if (nx >= 0 && nx < matrix.size() && ny >= 0 && ny < matrix[nx].size()) {
if (dfs(matrix, nx, ny, p + 1, str))
return true;
}
}
matrix[x][y] = tmp; // 还原
return false;
}
};