给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
[‘A’,‘B’,‘C’,‘E’],
[‘S’,‘F’,‘C’,‘S’],
[‘A’,‘D’,‘E’,‘E’]
]
给定 word = “ABCCED”, 返回 true.
给定 word = “SEE”, 返回 true.
给定 word = “ABCB”, 返回 false.
第一次的代码:
首先遍历board,找到一个字母与word的第一个字母相同的字母的位置,那么这个位置有可能产生符合题意的字符串。当遍历到合适的位置[i,j]时,将坐标输入函数ws(),判断该位置不否可行,可行直接返回True,否则继续遍历到下一个可行点。
对于ws函数,避免走到走过的位置,使用used来储存走过的位置[i.j]。在每个可行位置上,分别有4种情况,上下左右,则分别判断是否可行。ws函数返回的是bool,所以直接将函数放到判断条件上。最后,记得used.pop(),即回溯,若走过的点[i,j]当前可行,并且已经加入了used,但是判断当上下左右移动的时候,没有返回TRue,表明当前这个点不可行,则需要从used中弹出,从上一状态继续储存下一个可行点。若在某一个if成功返回True,则不会执行used.pop().
def ws(i,j,l1,l2,start,board,word,used):
if board[i][j]==word[start]:
used.append([i,j])
if start==len(word)-1:
return True
else: #4种情况
if i<l1-1 and [i+1,j] not in used and ws(i+1,j,l1,l2,start+1,board,word,used):
return True
if j<l2-1 and [i,j+1] not in used and ws(i,j+1,l1,l2,start+1,board,word,used):
return True
if i>0 and [i-1,j] not in used and ws(i-1,j,l1,l2,start+1,board,word,used):
return True
if j>0 and [i,j-1] not in used and ws(i,j-1,l1,l2,start+1,board,word,used):
return True
used.pop()#回溯,即若当前[i,j]是上一个点往左,不可行,出栈,加入上一个点往右的可行点。
else:
return False
l1=len(board)
l2=len(board[0])
for i in range(l1):
for j in range(l2):
if board[i][j]==word[0]:
if ws(i,j,l1,l2,0,board,word,[]):
return True
return False
288ms,排名69.7%,虽然慢,但是代码可读性较强,能简单理清楚以及更好的理解回溯,我不优化了。