【leetCode】Word Search II

题意：给定一组二维小写字母数组和一组单词序列，问单词序列里有多少个单词能从二维数组某个点开始，上下左右移动形成？

思路：单词序列用Trie树压缩，然后从树根开始，在二维数组里找单词即可。Trie树很好理解，适合于前缀重叠的情况。

代码：

class Solution {
public:
    struct Node{
        int end;
        Node *next[26];
        Node(){
            end=-1;
            for(int i=0;i<26;i++) next[i]=NULL;
        }
    };
    vector<bool> isGot;
    void check(vector<vector<char>>& board, vector<string>& words,vector<string> &ans,vector<vector<bool>> &isUsed,Node *trie,int i,int j)
    {
        if(trie->end!=-1&&isGot[trie->end]==false)
        {
            isGot[trie->end]=true;
            ans.push_back(words[trie->end]);
        }
        int d[4][2]={{0,1},{1,0},{0,-1},{-1,0}};
        int k,t,w,x,y;
        for(k=0;k<26;k++)
        {
            if(trie->next[k]!=NULL)
            {
                for(t=0;t<4;t++)
                {
                    x=i+d[t][0];
                    y=j+d[t][1];
                    if(x>=0&&x<board.size()&&y>=0&&y<board[0].size()&&board[x][y]==k+'a'&&isUsed[x][y]==false)
                    {
                        isUsed[x][y]=true;
                        check(board,words,ans,isUsed,trie->next[k],x,y);
                        isUsed[x][y]=false;
                    }
                }
            }
        }
    }
    vector<string> findWords(vector<vector<char>>& board, vector<string>& words) {
        vector<string> ans;
        Node *trie=new Node(),*temp;
        int i,j,k,m,n;
        m=board.size();
        if(m==0) return ans;
        n=board[0].size();
        vector<vector<bool>> isUsed(m,vector<bool>(n,false));
        isGot=vector<bool>(words.size(),false);
        for(i=0;i<words.size();i++)
        {
            temp=trie;
            for(j=0;j<words[i].size();j++)
            {
                if(temp->next[words[i][j]-'a']==NULL) temp->next[words[i][j]-'a']=new Node();
                temp=temp->next[words[i][j]-'a'];
            }
            temp->end=i;
        }
        temp=trie;
        for(i=0;i<26;i++)
        {
            if(temp->next[i]==NULL) continue;
            for(j=0;j<m;j++)
            {
                for(k=0;k<n;k++)
                {
                    if(board[j][k]==i+'a')
                    {
                        isUsed[j][k]=true;
                        check(board,words,ans,isUsed,trie->next[i],j,k);
                        isUsed[j][k]=false;
                    }
                }
            }
        }
        return ans;
    }
};