题目描述
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
给定 word = "ABCCED", 返回 true
给定 word = "SEE", 返回 true
给定 word = "ABCB", 返回 false
解题思路
这道题看懂了题意还是很简单的,即一个dfs回溯即可,中间记录走过的点的坐标
class Solution(object):
def __init__(self):
self.visit = []
def exist(self, board, word):
target = len(word)
m, n = len(board), len(board[0])
flag = False
def dfs(x, y, tempStr):
if len(tempStr) == 0:
return True
if x < 0 or x >= m or y < 0 or y >= n:
return False
if (x, y) in self.visit:
return False
temp = False
if board[x][y] == tempStr[0]:
self.visit.append((x, y))
temp = dfs(x - 1, y, tempStr[1: ]) or dfs(x + 1, y, tempStr[1: ]) or dfs(x, y - 1, tempStr[1: ]) or \
dfs(x, y + 1, tempStr[1: ])
self.visit.pop()
return temp
for i in range(len(board)):
for j in range(len(board[0])): #从每个坐标开始遍历
self.visit.clear() #每次从一个点开始时,清空已访问的点
if board[i][j] == word[0]:
flag = dfs(i, j, word)
if flag is True:
return flag
return flag
S = Solution()
print(S.exist([["A","B","C","E"],["S","F","E","S"],["A","D","E","E"]], "ABCESEEEFS"))
代码中self.visit.pop()这句话很重要,比如对于
[[A, B, C, E],
[S, F, E, S],
[A, D, E, E]]
这里遍历(0, 0)时,发现A与字符串起点相同,于是可以开始遍历,当向右移到C处时,比如先走下方的E,根据代码的执行过程,E对应(1, 2)应该加入到visit中,可是发现走不通,此时应该释放这个(1, 2)坐标,试想如果没有释放,那么C从坐标(0, 3)处的E开始走,根据待匹配结果,还必须经过(1, 2)处的E,如果没有释放,程序就会运行不出来,本来返回True的结果返回False。