题目:
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]
给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-search
思路:
先寻找字符串中首字母在数组中的位置,再在这个位置向四周寻找,字符串下一个字母的位置.以此类推.
注意,已寻找到的单词进行标记.
代码:
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
if not word :
return not board
m=len(board)
n=len(board[0])
#result=[]
flag=[[False for j in range(n)] for i in range(m)]
for m1 in range(m):
for n1 in range(n):
if board[m1][n1]==word[0]:
if self.search(board,0,m1,n1,word,flag):
return True
return False
def search(self,board,k,i,j,word,flag):
#result=False
if k==len(word):
return True
if i< 0 or i >=len(board) or j<0 or j>=len(board[0]) or flag[i][j] or board[i][j]!=word[k]:
return False
flag[i][j]=True
result = self.search(board, k+ 1, i + 1, j, word,flag) or\
self.search(board, k + 1, i - 1, j, word,flag) or\
self.search(board, k + 1, i, j + 1, word,flag) or\
self.search(board, k + 1, i, j - 1, word,flag)
flag[i][j]=False
return result