LeetCode算法题79:单词搜索解析

给定一个二维网格和一个单词,找出该单词是否存在于网格中。

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

示例:

board =
[
  ['A','B','C','E'],
  ['S','F','C','S'],
  ['A','D','E','E']
]

给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.

这个题就是一个迷宫题,一般就是使用DFS来进行搜索,不同的是这里不是一个开始点,要对每一个点都开始一次搜索。用一个数组来标记所有的地方是否被走过。具体思路看程序会更清楚一些。有意思的一点是,C++程序中如果使用bool类型来定义标记数组,那么就会很慢而且占用空间非常大,而使用int类型定义标记数组时,程序运行速度会很快而且占用空间会减少很多。
C++源代码:

class Solution {
public:
    bool exist(vector<vector<char>>& board, string word) {
        if(board.empty() || word.empty()) return false;
        int m = board.size(), n = board[0].size();
        vector<vector<int>> flag(m, vector<int>(n, 0));
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(search(board, word, 0, i, j, flag))
                    return true;
            }
        }
        return false;
    }
    
    bool search(vector<vector<char>>& board, string& word, int idx, int x, int y, vector<vector<int>>& flag){
        if(idx == word.size()) return true;
        int m = board.size(), n = board[0].size();
        if(x<0 || y<0 || x>=m || y>=n || flag[x][y] || board[x][y]!=word[idx]) return false;
        flag[x][y] = 1;
        bool res = search(board, word, idx+1, x+1, y, flag) ||
                    search(board, word, idx+1, x-1, y, flag) ||
                    search(board, word, idx+1, x, y+1, flag) ||
                    search(board, word, idx+1, x, y-1, flag);
        flag[x][y] = 0;
        return res;
    }
};

python3源代码:

class Solution:
    def exist(self, board, word):
        """
        :type board: List[List[str]]
        :type word: str
        :rtype: bool
        """
        if len(board)==0 or len(word)==0:
            return False
        m = len(board)
        n = len(board[0])
        flag = [[0 for j in range(n)] for i in range(m)]
        for i in range(m):
            for j in range(n):
                if self.search(board, word, 0, i, j, flag):
                    return True
        return False
    
    def search(self, board, word, idx, x, y, flag):
        if idx == len(word):
            return True
        m = len(board)
        n = len(board[0])
        if x<0 or y<0 or x>=m or y>=n or flag[x][y] or board[x][y]!=word[idx]:
            return False
        flag[x][y] = 1
        res = self.search(board, word, idx+1, x+1, y, flag) or self.search(board, word, idx+1, x-1, y, flag) or self.search(board, word, idx+1, x, y+1, flag) or self.search(board, word, idx+1, x, y-1, flag)
        flag[x][y] = 0
        return res
            

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/87719152