刷题日志：Leetcode 79 Word Search(Python)

题目描述

Given a 2D board and a word, find if the word exists in the grid.

The word can be constructed from letters of sequentially adjacent cell, where “adjacent” cells are those horizontally or vertically neighboring. The same letter cell may not be used more than once.

例子

board =
[
[‘A’,‘B’,‘C’,‘E’],
[‘S’,‘F’,‘C’,‘S’],
[‘A’,‘D’,‘E’,‘E’]
]
Given word = “ABCCED”, return true.
Given word = “SEE”, return true.
Given word = “ABCB”, return false.

解释

给一个矩阵，矩阵的每一个位置放一个字母。然后再给一个单词，从矩阵任意位置开始，每一步可以往上下左右走，但不能重复走，是否能有一条路径组成该单词。
比如第二个里SEE，从第二行最后一个S开始，先往下，再往左，便可以组成SEE，所以返回True。

思路

这个题我没有想过用暴力解，因为可能性实在太多了…
讲正经的，这是一道典型的深度优先搜索（DFS）。简单来说DFS就是一条路走到底，走不通了再退回交叉口换另一条路。一直到所有可能的点都走到（或找到符合条件的路径）。
用一个例子讲一下，还以SEE为例，从第二行最后一列的S开始，如果我们先往上走，可以看到第一个字母为E，符合条件。但此时如论你往哪里走，都无法找到符合条件的相邻的第二个E，此时认为是这条路走到底了。那就往回退，上次的分叉路口是S，那么我们从S往下走，找到E，再往左走，找到E，递归结束。具体的在代码中注释。

代码

class Solution(object):
    def exist(self, board, word):
        """
        :type board: List[List[str]]
        :type word: str
        :rtype: bool
        """
        #边界条件
        if not word:
            return True
        #遍历矩阵，如果遇到word中第一个字母，开始DFS。
        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] == word[0]:
                    if self.dfs(board, i, j, word[1:]):
                        return True
        return False
        
    #我们写一个辅助函数DFS，这样在主函数中调用起来比较简洁。
    #board, i, j为别是字母矩阵和当前位置的xy坐标
    #word是每次开始递归时目标单词剩余的单词。
    #比如SEE，在从S开始时，目标word就变成了EE，即word[1:]
    def dfs(self, board, i, j, word):
        #如果目标单词空了，就意味着已经找到了所有字母，返回True
        if not word:
            return True
        #在每一个位置，都可能有上下左右四个方向
        direction = [(1, 0), (0, 1), (-1, 0), (0, -1)]
        for x, y in direction:
            if 0 <= i + x < len(board) and 0 <= j + y < len(board[0]) and word[0] == board[i + x][j + y]:
                #这里先把当前位置替换成任意字符，只要不属于A-Z。
                #因为我们不能走回头路。
                tmp = board[i][j]
                board[i][j] = '@'
                #如果递归中找到符合条件的路径，返回True
                if self.dfs(board, i + x, j + y, word[1:]):
                    return True
                #在每次递归结束后（即每条路走到底后），把之前的特殊字符替换回来。
                board[i][j] = tmp
        #如果四个方向都没有正确字母，说明这条路走不通，返回False。
        return False