题目描述:
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
给定 word = “ABCCED”, 返回 true.
给定 word = “SEE”, 返回 true.
给定 word = “ABCB”, 返回 false.
解题思路:
- 按照从左到右的顺序查找字符串word中的元素,找到后查找下一个元素的方向是上下左右四个方向,按照深度优先dfs的方式进行遍历
- 需要注意边界条件和查找跳出条件
代码:
class Solution(object):
def dfs(self, board, word, d, x, y):
# 边界条件,不满足即跳出
if(x < 0 or x >= len(board) or y < 0 or y >= len(board[0]) or board[x][y] != word[d]):
return False
if(d==len(word)-1): # 字符串word中所有元素都被查找到,说明已经找到了,返回true
return True
curr = board[x][y]
board[x][y] = 0 # 将当前位置的值设置成0,避免查找到之前走过的路径
# 上下左右四个方向进行查找
if(self.dfs(board,word,d+1,x+1,y) or self.dfs(board,word,d+1,x-1,y) or self.dfs(board,word,d+1,x,y+1) or self.dfs(board,word,d+1,x,y-1)):
return True
board[x][y] = curr # 恢复当前位置的值
return False
def exist(self, board, word):
if (not board):
return False
m = len(board)
n = len(board[0])
for i in range(m):
for j in range(n):
if self.dfs(board, word, 0, i, j):
return True
return False
s = Solution()
board = [
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
word = "ABCCED"
print(s.exist(board, word))
参考链接:
leetcode–79–单词搜索