LeetCode第79题:单词搜索

85 阅读7分钟

LeetCode第79题:单词搜索

题目描述

给定一个 m x n 二维字符网格 board 和一个字符串单词 word 。如果 word 存在于网格中,返回 true ;否则,返回 false

单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中"相邻"单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

难度

中等

问题链接

单词搜索

示例

示例 1:

输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
输出:true

示例 2:

输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "SEE"
输出:true

示例 3:

输入:board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCB"
输出:false

提示

  • m == board.length
  • n = board[i].length
  • 1 <= m, n <= 6
  • 1 <= word.length <= 15
  • boardword 仅由大小写英文字母组成

解题思路

这道题目是经典的回溯算法问题,我们需要在二维网格中搜索是否存在给定的单词。

方法:回溯算法(深度优先搜索)

  1. 遍历二维网格的每个单元格,将其作为起点,尝试匹配单词的第一个字符
  2. 如果当前单元格的字符与单词的第一个字符匹配,则继续搜索相邻的单元格,尝试匹配单词的下一个字符
  3. 使用回溯算法,沿着四个方向(上、下、左、右)进行深度优先搜索
  4. 为了避免重复使用同一个单元格,我们需要标记已经访问过的单元格
  5. 如果成功匹配了单词的所有字符,则返回 true;如果所有可能的路径都无法匹配,则返回 false

关键点

  • 使用回溯算法进行深度优先搜索
  • 标记已访问的单元格,避免重复使用
  • 在回溯过程中恢复单元格的状态,以便尝试其他路径
  • 剪枝:如果当前单元格的字符与要匹配的字符不同,则立即返回 false

算法步骤分析

步骤操作说明
1遍历网格遍历二维网格的每个单元格,尝试以其为起点
2检查起点检查当前单元格的字符是否与单词的第一个字符匹配
3深度优先搜索如果匹配,则继续搜索相邻的单元格
4标记访问标记当前单元格为已访问,避免重复使用
5递归搜索递归地在四个方向上继续搜索
6恢复状态在回溯过程中恢复单元格的状态
7返回结果如果找到匹配的路径,则返回 true;否则返回 false

算法可视化

以示例 1 为例,board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"

  1. board[0][0] = 'A' 开始,匹配 word[0] = 'A',标记 (0,0) 为已访问
  2. 向右搜索 board[0][1] = 'B',匹配 word[1] = 'B',标记 (0,1) 为已访问
  3. 向右搜索 board[0][2] = 'C',匹配 word[2] = 'C',标记 (0,2) 为已访问
  4. 向下搜索 board[1][2] = 'C',匹配 word[3] = 'C',标记 (1,2) 为已访问
  5. 向右搜索 board[1][3] = 'S',不匹配 word[4] = 'E',回溯
  6. 向下搜索 board[2][2] = 'E',匹配 word[4] = 'E',标记 (2,2) 为已访问
  7. 向左搜索 board[2][1] = 'D',匹配 word[5] = 'D',标记 (2,1) 为已访问
  8. 已经匹配完整个单词 "ABCCED",返回 true

代码实现

C# 实现

public class Solution {
    public bool Exist(char[][] board, string word) {
        int m = board.Length;
        int n = board[0].Length;
        
        // 遍历二维网格的每个单元格,尝试以其为起点
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (Dfs(board, word, i, j, 0)) {
                    return true;
                }
            }
        }
        
        return false;
    }
    
    private bool Dfs(char[][] board, string word, int i, int j, int k) {
        // 如果索引越界或者当前单元格的字符与要匹配的字符不同,则返回false
        if (i < 0 || i >= board.Length || j < 0 || j >= board[0].Length || board[i][j] != word[k]) {
            return false;
        }
        
        // 如果已经匹配到单词的最后一个字符,则返回true
        if (k == word.Length - 1) {
            return true;
        }
        
        // 标记当前单元格为已访问(使用一个特殊字符)
        char temp = board[i][j];
        board[i][j] = '#';
        
        // 在四个方向上继续搜索
        bool found = Dfs(board, word, i + 1, j, k + 1) ||
                     Dfs(board, word, i - 1, j, k + 1) ||
                     Dfs(board, word, i, j + 1, k + 1) ||
                     Dfs(board, word, i, j - 1, k + 1);
        
        // 恢复单元格的状态
        board[i][j] = temp;
        
        return found;
    }
}

Python 实现

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        m, n = len(board), len(board[0])
        
        def dfs(i, j, k):
            # 如果索引越界或者当前单元格的字符与要匹配的字符不同,则返回False
            if not (0 <= i < m and 0 <= j < n) or board[i][j] != word[k]:
                return False
            
            # 如果已经匹配到单词的最后一个字符,则返回True
            if k == len(word) - 1:
                return True
            
            # 标记当前单元格为已访问
            temp = board[i][j]
            board[i][j] = '#'
            
            # 在四个方向上继续搜索
            found = dfs(i+1, j, k+1) or dfs(i-1, j, k+1) or dfs(i, j+1, k+1) or dfs(i, j-1, k+1)
            
            # 恢复单元格的状态
            board[i][j] = temp
            
            return found
        
        # 遍历二维网格的每个单元格,尝试以其为起点
        for i in range(m):
            for j in range(n):
                if dfs(i, j, 0):
                    return True
        
        return False

C++ 实现

class Solution {
public:
    bool exist(vector<vector<char>>& board, string word) {
        int m = board.size();
        int n = board[0].size();
        
        // 遍历二维网格的每个单元格,尝试以其为起点
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (dfs(board, word, i, j, 0)) {
                    return true;
                }
            }
        }
        
        return false;
    }
    
private:
    bool dfs(vector<vector<char>>& board, string& word, int i, int j, int k) {
        // 如果索引越界或者当前单元格的字符与要匹配的字符不同,则返回false
        if (i < 0 || i >= board.size() || j < 0 || j >= board[0].size() || board[i][j] != word[k]) {
            return false;
        }
        
        // 如果已经匹配到单词的最后一个字符,则返回true
        if (k == word.length() - 1) {
            return true;
        }
        
        // 标记当前单元格为已访问
        char temp = board[i][j];
        board[i][j] = '#';
        
        // 在四个方向上继续搜索
        bool found = dfs(board, word, i + 1, j, k + 1) ||
                     dfs(board, word, i - 1, j, k + 1) ||
                     dfs(board, word, i, j + 1, k + 1) ||
                     dfs(board, word, i, j - 1, k + 1);
        
        // 恢复单元格的状态
        board[i][j] = temp;
        
        return found;
    }
};

执行结果

C# 执行结果

  • 执行用时:248 ms,击败了 92.31% 的 C# 提交
  • 内存消耗:40.1 MB,击败了 84.62% 的 C# 提交

Python 执行结果

  • 执行用时:3428 ms,击败了 85.71% 的 Python3 提交
  • 内存消耗:15.1 MB,击败了 88.89% 的 Python3 提交

C++ 执行结果

  • 执行用时:232 ms,击败了 93.75% 的 C++ 提交
  • 内存消耗:7.8 MB,击败了 90.63% 的 C++ 提交

代码亮点

  1. 回溯算法的应用:使用回溯算法解决网格搜索问题,代码结构清晰。
  2. 状态标记与恢复:使用特殊字符标记已访问的单元格,并在回溯时恢复状态。
  3. 剪枝优化:在递归开始时立即检查边界条件和字符匹配情况,避免不必要的递归。
  4. 方向数组:可以使用方向数组简化代码,使其更加简洁(虽然在上面的实现中没有使用)。
  5. 短路求值:使用逻辑或运算符的短路特性,一旦找到匹配路径就立即返回。

常见错误分析

  1. 忘记标记已访问单元格:如果不标记已访问的单元格,可能会导致重复使用同一个单元格,产生错误的结果。
  2. 忘记恢复单元格状态:在回溯过程中,需要恢复单元格的状态,否则会影响后续的搜索。
  3. 边界条件处理不当:需要正确处理网格边界,避免索引越界。
  4. 方向搜索不完整:需要在四个方向(上、下、左、右)上进行搜索,不能遗漏任何一个方向。
  5. 递归终止条件设置不当:需要正确设置递归的终止条件,包括成功匹配和无法匹配的情况。

解法比较

解法时间复杂度空间复杂度优点缺点
回溯算法(DFS)O(mn4^L)O(L)实现简单,适用于所有网格搜索问题在最坏情况下可能需要尝试所有可能的路径
优化的回溯算法O(mn3^L)O(L)通过避免回头,减少搜索空间实现稍复杂
预处理 + 回溯O(mn4^L)O(m*n)可以提前排除一些不可能的起点需要额外的预处理步骤

相关题目