LeetCode 79. Word Search

题目

思路

1. 主函数内，两重循环遍历二维数组，调用回溯的递归函数，选择每个元素作为匹配的起始位置，一旦找到就返回true。
2. 递归函数的作用从当前位置开始board[startx][starty]寻找word[index,:]，结束条件是index取到最后一位时，当前位置的值等于要找的字母，返回true；递归过程要先判断当前位置的值是否等于要找的index位置的值，如果是再递归调用上下左右寻找下一个元素，否则直接返回false
3. 使用一个自定义二维数组+一个4次循环来实现上下左右的遍历
4. 判断是否超出边界，使用一个函数实现，比较简洁
5. 如果上下左右都不对，要回退，设置visited为false

代码

class Solution:
    def exist(self, board, word):
        m = len(board)
        n = len(board[0])
        visited = [[0 for i in range(n)] for j in range(m)]
        for i in range(m):
            for j in range(n):
                if self.find(board, word, 0, i, j, visited): 
                    return True
        return False

    def inarea(self, x,y,m,n):
        if (x<m and y <n and x>=0 and y>=0):
            return True
        return False

    def find(self, board, word, index, startx, starty, visited):
        m = len(board)
        n = len(board[0])
        d = [[1,0],[0,1],[-1,0],[0,-1]]
        if index == len(word)-1:
            return board[startx][starty] == word[index]
        if board[startx][starty] == word[index]:
            visited[startx][starty] = 1
            for i in range(4):
                x = startx + d[i][0]
                y = starty + d[i][1]
                if self.inarea(x,y,m,n)==True and visited[x][y]==0 and self.find(board, word, index+1, x, y, visited) == True: 
                    return True
            visited[startx][starty] = 0
        return False