zoukankan      html  css  js  c++  java
  • [LeetCode] 79. Word Search 词语搜索

    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.

    For example,
    Given board =

    [
      ["ABCE"],
      ["SFCS"],
      ["ADEE"]
    ]
    

    word = "ABCCED", -> returns true,
    word = "SEE", -> returns true,
    word = "ABCB", -> returns false.

    这道题是典型的深度优先遍历 DFS 的应用,原二维数组就像是一个迷宫,可以上下左右四个方向行走,我们以二维数组中每一个数都作为起点和给定字符串做匹配,我们还需要一个和原数组等大小的 visited 数组,是 bool 型的,用来记录当前位置是否已经被访问过,因为题目要求一个 cell 只能被访问一次。如果二维数组 board 的当前字符和目标字符串 word 对应的字符相等,则对其上下左右四个邻字符分别调用 DFS 的递归函数,只要有一个返回 true,那么就表示可以找到对应的字符串,否则就不能找到,具体看代码实现如下:

    解法一:

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

    我们还可以不用 visited 数组,直接对 board 数组进行修改,将其遍历过的位置改为井号,记得递归调用完后需要恢复之前的状态,参见代码如下:

    解法二:

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

    Github 同步地址:

    https://github.com/grandyang/leetcode/issues/79

    类似题目:

    Word Search II

    参考资料:

    https://leetcode.com/problems/word-search/

    https://leetcode.com/problems/word-search/discuss/27658/Accepted-very-short-Java-solution.-No-additional-space.

    https://leetcode.com/problems/word-search/discuss/27829/C++-backtracking-solution-without-extra-data-structure

    LeetCode All in One 题目讲解汇总(持续更新中...)

  • 相关阅读:
    unittest模块小结
    自定义日志类
    配置文件的简单操作
    自动化框架搭建(一)
    openpyxl一点心得
    关于过滤字符中不同元素小结
    关于传入列表出现元素次数排序题目的一点心得
    Python中*和**的作用(课堂小结)
    Python代码的人机大战(循环嵌套)
    最后期限阅读笔记02
  • 原文地址:https://www.cnblogs.com/grandyang/p/4332313.html
Copyright © 2011-2022 走看看