请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径。
[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
示例:
输入:board = [[“A”,“B”,“C”,“E”],[“S”,“F”,“C”,“S”],[“A”,“D”,“E”,“E”]], word = “ABCCED”
输出:true
解题思路:
本问题是典型的矩阵搜索问题,可使用 深度优先搜索(DFS)+ 剪枝 解决。
算法原理:
- 深度优先搜索: 可以理解为暴力法遍历矩阵中所有字符串可能性。DFS 通过递归,先朝一个方向搜到底,再回溯至上个节点,沿另一个方向搜索,以此类推。
- 剪枝: 在搜索中,遇到
这条路不可能和目标字符串匹配成功
的情况(例如:此矩阵元素和目标字符不同、此元素已被访问),则应立即返回,称之为可行性剪枝
。
算法剖析:
- 递归参数: 当前元素在矩阵
board
中的行列索引i
和j
,当前目标字符在word
中的索引k
。 - 终止条件:
- 返回 falsefalsefalse : ① 行或列索引越界 或 ② 当前矩阵元素与目标字符不同 或 ③ 当前矩阵元素已访问过 (③ 可合并至 ② ) 。
- 返回 truetruetrue : 字符串
word
已全部匹配,即k = len(word) - 1
。
- 递推工作:
- 标记当前矩阵元素: 将
board[i][j]
值暂存于变量tmp
,并修改为字符'/'
,代表此元素已访问过,防止之后搜索时重复访问。 - 搜索下一单元格: 朝当前元素的 上、下、左、右 四个方向开启下层递归,使用
或
连接 (代表只需一条可行路径) ,并记录结果至res
。 - 还原当前矩阵元素: 将
tmp
暂存值还原至board[i][j]
元素。
- 标记当前矩阵元素: 将
- 回溯返回值: 返回
res
,代表是否搜索到目标字符串。
图解中,从每个节点 DFS 的顺序为:下、上、右、左。
![](https://i-blog.csdnimg.cn/blog_migrate/ba2a397ad982c11e04103bb5781c3951.png)
![](https://i-blog.csdnimg.cn/blog_migrate/bc3a1ca798ac11815e0fa11b06d47a92.png)
![](https://i-blog.csdnimg.cn/blog_migrate/cbb4c160f6843890084c0471f9c4fa07.png)
![](https://i-blog.csdnimg.cn/blog_migrate/d687cf92f7ef515afcfec4f3df741086.png)
![](https://i-blog.csdnimg.cn/blog_migrate/83078590197baf2648a80552e3a79b15.png)
![](https://i-blog.csdnimg.cn/blog_migrate/f26ebe98863810d234c14fc407c59a35.png)
![](https://i-blog.csdnimg.cn/blog_migrate/765e51a1eaf589915fa16a7a0c7fd90b.png)
![](https://i-blog.csdnimg.cn/blog_migrate/60f378a0db8c2a9a0ff4247361690d01.png)
![](https://i-blog.csdnimg.cn/blog_migrate/6386aa44fa85e6db628823f43814c8fb.png)
![](https://i-blog.csdnimg.cn/blog_migrate/9f1e3da7f8f35d24b3010aca5872827e.png)
![](https://i-blog.csdnimg.cn/blog_migrate/3943fdf77e3ecb249f5a6f840b062ad3.png)
![](https://i-blog.csdnimg.cn/blog_migrate/ad3e030b2959503ec1e778e2b43997c6.png)
![](https://i-blog.csdnimg.cn/blog_migrate/cd503b299d3961af8c631dd4c7a73ee8.png)
![](https://i-blog.csdnimg.cn/blog_migrate/bd643f3323a9d6d686472ab49267aac9.png)
![](https://i-blog.csdnimg.cn/blog_migrate/e57ea6463e4926c09b9308105de2e22c.png)
![](https://i-blog.csdnimg.cn/blog_migrate/ad1b933ccbf6b31d5cc07436d7fd3d60.png)
![](https://i-blog.csdnimg.cn/blog_migrate/c0b806b22377c21da3fa5de95ddbd470.png)
![](https://i-blog.csdnimg.cn/blog_migrate/e703342e785fae87012e99413081d3cb.png)
复杂度分析:
M,NM, NM,N 分别为矩阵行列大小, KKK 为字符串
word
长度。
- 时间复杂度 O(3KMN)O(3^KMN)O(3KMN) : 最差情况下,需要遍历矩阵中长度为 KKK 字符串的所有方案,时间复杂度为 O(3K)O(3^K)O(3K);矩阵中共有 MNMNMN 个起点,时间复杂度为 O(MN)O(MN)O(MN) 。
- 方案数计算: 设字符串长度为 KKK ,搜索中每个字符有上、下、左、右四个方向可以选择,舍弃回头(上个字符)的方向,剩下 333 种选择,因此方案数的复杂度为 O(3K)O(3^K)O(3K) 。
- 空间复杂度 O(K)O(K)O(K) : 搜索过程中的递归深度不超过 KKK ,因此系统因函数调用累计使用的栈空间占用 O(K)O(K)O(K) (因为函数返回后,系统调用的栈空间会释放)。最坏情况下 K=MNK = MNK=MN ,递归深度为 MNMNMN ,此时系统栈使用 O(MN)O(MN)O(MN) 的额外空间。
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
def dfs(i, j, k):
if not 0<=i<len(board) or not 0<=j<len(board[0]) or board[i][j] != word[k]:
return False
if k == len(word)-1:
return True
temp, board[i][j] = board[i][j], '/'
res = dfs(i + 1, j, k + 1) or dfs(i - 1, j, k + 1) or dfs(i, j+1, k + 1) or dfs(i, j-1, k + 1)
board[i][j] = temp
return res
for i in range(len(board)):
for j in range(len(board[0])):
if dfs(i,j,0):
return True
return False
其中图来自https://leetcode-cn.com/problems/ju-zhen-zhong-de-lu-jing-lcof/solution/mian-shi-ti-12-ju-zhen-zhong-de-lu-jing-shen-du-yo/
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ju-zhen-zhong-de-lu-jing-lcof