题意:给定一组二维小写字母数组和一组单词序列,问单词序列里有多少个单词能从二维数组某个点开始,上下左右移动形成?
思路:单词序列用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;
}
};