1 题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。
[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
示例1:
输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
输出:true
示例2:
输入:board = [["a","b"],["c","d"]], word = "abcd"
输出:false
2 解题思路
本题是典型的矩阵搜索问题,可使用深度优先搜索(DFS)+ 剪枝 解决。
- 深度优先搜索: 可以理解为暴力法遍历矩阵中所有字符串可能性。DFS通过递归,先朝一个方向搜到底,再回溯至上个节点,沿另一个方向搜索,以此类推。
- 剪枝: 在搜索中,遇到这条路不可能和目标字符串匹配成功的情况(例如:此矩阵元素和目标字符不同、此元素已被访问),则应立即返回,称之为可行性剪枝。
算法流程:
- 递归参数:当前元素在矩阵board中的行列索引i和j,当前目标字符在word中的索引k。
- 终止条件:
- 返回false:(1)行或列索引越界 或(2)当前矩阵元素与目标字符不同 或(3)当前矩阵元素已访问过((3)可合并至(2))。
- 返回true:
k=len(word)-1
,即字符串word已全部匹配。
- 递推工作:
- 标记当前矩阵元素:将board[i][j]修改为空字符‘\0’,代表此元素已访问过,防止之后搜索时重复访问。
- 搜索下一单元格:朝当前元素的上、下、左、右四个方向开启下层递归,使用或连接(代表只需要找到一条可行路径就直接返回,不再做后续DFS),并记录结果至res。
- 还原当前矩阵元素:将board[i][j]元素还原至初始值,即word[k]。
- 返回值:返回布尔值res,代表是否搜索到目标字符串。
使用空字符做标记是为了防止标记字符与矩阵原有字符重复。当存在重复时,此算法会将矩阵原有字符认作标记字符,从而出现错误。
class Solution {
public boolean exist(char[][] board, String word) {
char[] words = word.toCharArray();
for (int i = 0;i < board.length;i++) {
for (int j = 0;j < board[0].length;j++) {
if (dfs(board,words,i,j,0)) return true;
}
}
return false;
}
private boolean dfs(char[][] board, char[] words, int i, int j, int k) {
if (i >= board.length || i < 0 || j >= board[0].length || j < 0 || board[i][j] != words[k]) return false;
if (k == words.length - 1) return true;
board[i][j] = '\0';
boolean res = dfs(board,words,i,j+1,k+1) || dfs(board,words,i,j-1,k+1) || dfs(board,words,i-1,j,k+1) || dfs(board,words,i+1,j,k+1);
board[i][j] = words[k];
return res;
}
}
复杂度分析:
M,N 分别为矩阵行列大小,K为字符串 word 长度。
- 时间复杂度 O(
3
K
3^K
3KMN): 最差情况下,需要遍历矩阵中长度为K字符串的所有方案,时间复杂度为O(
3
K
3^K
3K);矩阵中共有MN个起点,时间复杂度为O(MN)。
- 方案数计算: 设字符串长度为K,搜索中每个字符有上、下、左、右四个方向可以选择,舍弃回头(上个字符)的方向,剩下3种选择,因此方案数的复杂度为O( 3 K 3^K 3K)。
- 空间复杂度O(K):搜索过程中的递归深度不超过K,因此系统因函数调用累计使用的栈空间占用O(K)(因为函数返回后,系统调用的栈空间会释放)。最坏情况下K=MN,递归深度为MN,此时系统栈使用O(MN)的额外空间。