弄个模版也能花一天。。。。太佩服我自己了。。。。root的初始化都没弄好。。
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
#define val 10005
using namespace std;
struct Dictree
{
int cnt;
Dictree *fail,*next[26];
}*root;
queue<Dictree*> q;
int n,ans;
char str[val*100],s[val*100];
void build_tree();
void insert(char *,int);
void set_fail();
void ac_auto(char *);
void empty_tree(Dictree *);
int main()
{
int t;
for(scanf("%d",&t);t;t--)
{
scanf("%d",&n);
getchar();
build_tree();
set_fail();
gets(str);
ans=0;
ac_auto(str);
printf("%d\n",ans);
empty_tree(root);
}
return 0;
}
void build_tree()
{
int i;
root=(Dictree*)malloc(sizeof(Dictree));
root->cnt=0;
memset(root->next,NULL,sizeof(root->next));
for(i=0;i<n;i++)
{
gets(s);
insert(s,i);
}
}
void insert(char *s,int no)
{
int i;
struct Dictree *p,*t;
p=root;
for(i=0;s[i];i++)
{
if(p->next[s[i]-'a']==NULL)
{
t=(Dictree*)malloc(sizeof(Dictree));
memset(t->next,NULL,sizeof(t->next));
t->cnt=0;
t->fail=NULL;
p->next[s[i]-'a']=t;
}
p=p->next[s[i]-'a'];
}
p->cnt++;
}
void set_fail()
{
int i;
Dictree *p,*temp;
root->fail=NULL;
q.push(root);
while(!q.empty())
{
p=q.front();
q.pop();
for(i=0;i<26;i++)
{
if(p->next[i])
{
if(p==root) p->next[i]->fail=root;
else
{
temp=p->fail;//从父亲的fail开始寻找
while(temp)
{
if(temp->next[i])//儿子中没有该字母,继续寻找
{
p->next[i]->fail=temp->next[i];
break;
}
temp=temp->fail;
}
if(!temp) p->next[i]->fail=root;
}
q.push(p->next[i]);
}
}
}
}
void ac_auto(char *s)
{
int i,index;
Dictree *p,*temp;
p=root;
for(i=0;s[i];i++)
{
index=s[i]-'a';
while(p!=root)
{
if(p->next[index]) break;
p=p->fail;
}
temp=p=p->next[index];
if(p==NULL) temp=p=root;
while(temp)//找到了一个,继续找,但不要移动原来的位置
{
if(temp->cnt>=0)
{
ans+=temp->cnt;
temp->cnt=-1;
}
else break;
temp=temp->fail;
}
}
}
void empty_tree(Dictree *p)
{
int i;
for(i=0;i<26;i++)
if(p->next[i])
empty_tree(p->next[i]);
free(p);
}