Trie就是把n个字符串建成一个树,树的每个节点都是一个字母,然后字符串的最后一个节点,我们会在Trie中对应的节点处打上标记,然后从根节点到每个标记点的路径上的字母依次排开就是n个字符串中的一个。
是不是很浅显?
稍稍举个例子:
n=5
he
his
hers
abcx
bc
那么我们建的Trie应该是这样的
root
/ | \
h b a
/ \ | \
*e i *c b
| | |
r *s c
| |
*s *x
其中打*的就是标记点,即一个单词的结尾。
至于建Trie,就弄个链式前向星就好了,
设ch[u][c]表示u号节点的字母ASCII码值-1为c的孩子的编号,所以0<=c<26
那么在上面的例子中,h点的编号为1,e点的编号为4,所以ch[1][4]=4,
插入时就一个字母依次的,如果当前ch[u][c]为空,就新设一个点,最后在u打上标记。
查找时也是一个字母依次的,如果当前ch[u][c]为空,查找失败,否则把u调到ch[u][c],如果最后一个字母也找到了,再看看当前点是否有标记,没有标记则查找失败。反之,则查找成功。
如果没怎么看懂,就结合代码看看,其实很容易理解的。
以洛谷P2292,HNOI2004L语言为例。
一个Trie+很水的dp,时限卡的有点紧。
参考代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e6+10;
int n,m,len=0;
char P[N];
bool dp[N];
struct Trie{
int ch[211][26],val[211],NodeCnt;
inline void clear(){
memset(ch,0,sizeof(ch));
memset(val,0,sizeof(val));
NodeCnt=0;
}
inline void insert(char *s){
int u=0,l=strlen(s);
len=max(len,l);
for (int i=0;i<l;i++){
int c=s[i]-'a';
if (!ch[u][c])ch[u][c]=++NodeCnt;
u=ch[u][c];
}
val[u]=1;
}
inline bool find(int st,int en){
int u=0;
for (int i=st;i<=en;i++){
int c=P[i]-'a';
if (!ch[u][c])return 0;
u=ch[u][c];
}
return val[u];
}
}T;
int main(){
scanf("%d%d",&n,&m);
T.clear();
for (int i=1;i<=n;i++){
scanf("%s",P);
T.insert(P);
}
for (int i=1;i<=m;i++){
scanf("%s",P);
memset(dp,0,sizeof(dp));
int l=strlen(P),ans=0;
for (int j=0;j<l;j++)
for (int k=max(j-len,-1);k<=j;k++)
if ((k==-1 || dp[k]) && T.find(k+1,j)){
dp[j]=1;ans=j+1;break;
}
printf("%d\n",ans);
}
return 0;
}
裸的Trie题并不多,常常是做数据结构优化dp,贪心什么的,并不是核心。