Word Search II（c++单词搜索 II）

最新推荐文章于 2024-07-17 20:43:45 发布

coolsunxu

最新推荐文章于 2024-07-17 20:43:45 发布

阅读量111

点赞数

分类专栏： C++ LeetCode 文章标签： c++ leetcode 回溯前缀树

本文链接：https://blog.csdn.net/coolsunxu/article/details/119238444

版权

C++ 同时被 2 个专栏收录

735 篇文章 17 订阅

订阅专栏

LeetCode

459 篇文章 1 订阅

订阅专栏

（1）回溯+前缀树


class Trie {
public:
    vector<Trie*> v;
    bool isend;
public:
    
    Trie():v(26),isend(false) {
    }

    void insert(string word) {
        Trie* node=this;
        for(int i=0;i<word.length();i++) {
            int index=word[i]-'a';
            if(node->v[index]==nullptr) {
                node->v[index]=new Trie();
            }
            node=node->v[index];
        }
        node->isend=true;
    }
};
 
class Solution {
private:
    vector<vector<int>> d={{0,1},{0,-1},{1,0},{-1,0}};
    int m,n;
public:
    void helper(vector<vector<char>>& board, int x, int y, Trie* root, string word, set<string>& result){
        
        if(root->v[board[x][y]-'a'] != nullptr){
            word=word+board[x][y];
            root=root->v[board[x][y]-'a']; 
            if(root->isend) result.insert(word);
            char c=board[x][y];
            board[x][y]=' ';
            for(int i=0;i<d.size();i++) {
                int x1=x+d[i][0];
                int y1=y+d[i][1];
                if(x1>=0 && x1<m && y1>=0 && y1<n && board[x1][y1]!=' ') {
                    helper(board, x1, y1, root, word, result);
                }
            }
            board[x][y]=c;  
        }
        return;
    }

    vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
        Trie* trie = new Trie();
        for(auto &&w:words) {
            trie->insert(w);
        }
        
        m=board.size(),n=board[0].size();
        set<string> result_set;
        for(int i=0; i<m; i++)
            for(int j=0; j<n; j++)
                helper(board, i, j, trie, "", result_set);
        
        vector<string> result;
        for(auto it:result_set) result.push_back(it);
        return result;        
    }
};