212. Word Search II-优快云博客

本文介绍了一种解决二维板寻字游戏问题的高效算法，使用Trie树结合深度优先搜索（DFS）的方法，从给定的单词列表中找出所有能在二维字符板上构成的单词。文章详细解释了如何构建Trie树以及如何通过DFS遍历二维板来查找匹配的单词。

Given a 2D board and a list of words from the dictionary, find all words in the board.

Each word must 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 in a word.

Example:

Input: 
words = ["oath", "pea", "eat", "rain"] and board =
[
  ['o','a','a','n'],
  ['e','t','a','e'],
  ['i','h','k','r'],
  ['i','f','l','v']
]

Output: ["eat", "oath"]

Note:
You may assume that all inputs are consist of lowercase letters a-z.

Approach #1: Trie + DFS. [C++]

class Trie{
public:
    Trie *children[26];         // pointer to its substring starting with 'a' to 'z'
    bool leaf;      // if the node is a leaf, or if there is a word stopping at here
    int idx;        // if it is a leaf, the string index of the array words
    Trie() {
        this->leaf = false;
        this->idx = 0;
        fill_n(this->children, 26, nullptr);
    }
};


class Solution {
public:
    void insertWords(Trie *root, vector<string>& words, int idx) {
        int pos = 0, len = words[idx].size();
        while(pos < len) {
            if (nullptr == root->children[words[idx][pos]-'a'])
                root->children[words[idx][pos]-'a'] = new Trie();
            root = root->children[words[idx][pos++]-'a'];
        }
        root->leaf = true;
        root->idx = idx;
    }
    
    Trie *buildTrie(vector<string>& words) {
        Trie *root = new Trie();
        int i;
        for (i = 0; i < words.size(); ++i)
            insertWords(root, words, i);
        return root;
    }
    
    void checkWords(vector<vector<char>>& board, int i, int j, int row, 
                    int col, Trie *root, vector<string>& res, vector<string>& words) {
        char temp;
        if (board[i][j] == 'X') return ;        // visited before;
        if (nullptr == root->children[board[i][j]-'a']) return;     // no string with such prefix;
        else {
            temp = board[i][j];
            if (root->children[temp-'a']->leaf) {       // if it is a leaf
                res.push_back(words[root->children[temp-'a']->idx]);
                root->children[temp-'a']->leaf = false;     // set to false to indicate that we found it already
            }
            board[i][j] = 'X';      // mark the current position as visited
            // check all the possible neighbors
            if (i > 0) checkWords(board, i - 1, j, row, col, root->children[temp-'a'], res, words);
            if ((i+1) < row) checkWords(board, i + 1, j, row, col, root->children[temp-'a'], res, words);
            if (j > 0) checkWords(board, i, j - 1, row, col, root->children[temp-'a'], res, words);
            if ((j+1) < col) checkWords(board, i, j + 1, row, col, root->children[temp-'a'], res, words);
            board[i][j] = temp;
        }
    }
    
    vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
        vector<string> res;
        int row = board.size();
        if (0 == row) return res;
        int col = board[0].size();
        if (0 == col) return res;
        int wordCount = words.size();
        if (0 == wordCount) return res;
        
        Trie *root = buildTrie(words);
        
        int i, j;
        for (i = 0; i < row; ++i) {
            for (j = 0; j < col && wordCount > res.size(); ++j) {
                checkWords(board, i, j, row, col, root, res, words);
            }
        }
        return res;
    }
};

转载于:https://www.cnblogs.com/ruruozhenhao/p/10349498.html