题目描述:
题解:回溯
参考LeetCode 79 Word Search (Python) - 小明MaxMing
1.定义函数dfs,输入为一个坐标(posx,posy),需要匹配的字符串s和当前已经匹配到的下标idx,如果idx=len(s)说明已经完成了匹配,返回true,如果(posx,posy)超出界限或者与下一个需要匹配的s中的字符不同则返回,否则对当前位置的上下左右四个相邻位置调用dfs。
2.对输入board的每个位置进行搜索,如果找到一个位置与s第一个字符相等则调用dfs。
需要注意的地方:
1.在对相邻的四个位置调用dfs的时候,可能会对之前搜索过的位置进行重复搜索,所以定义一个visited数组,全部初始化为0,搜索过的对应位置值设置为1.
2.用idx记录当前匹配到的位置,不修改需要匹配的字符串。
3.在对(posx,posy)调用dfs前将该位置visited设为1,调用结束后重新修改为0,对应回溯的过程。
4.对输入board每个位置(i,j)进行判断,如果board[i][j]=word[0],则对该位置调用dfs,注意此时idx从1开始而不是从0,因为进入dfs就说明第一个字符已经匹配成功了。
class Solution(object): def exist(self, board, word): m = len(board) n = len(board[0]) visited = [[0 for j in range(n)]for i in range(m)] directions = [(-1,0),(1,0),(0,-1),(0,1)] def dfs(posx,posy,idx): if idx==len(word): return True for dx,dy in directions: newx = posx+dx newy = posy+dy if 0<=newx<m and 0<=newy<n and visited[newx][newy]==0 and board[newx][newy]==word[idx]: visited[newx][newy]=1 if dfs(newx,newy,idx+1): return True visited[newx][newy]=0 return False for i in range(m): for j in range(n): if board[i][j]==word[0]: visited[i][j]=1 if dfs(i,j,1): return True visited[i][j]=0 return False