hdu 2222 AC自动机模板题

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
#define maxn 241000
#define sigma_size 26
char s[60], t[maxn*5];
struct ac
{
int ch[maxn][sigma_size];
int val[maxn], f[maxn], last[maxn];
int next;
void init()
{
next = 0;
memset(ch[0], 0, sizeof(ch[0]));
memset(val, 0, sizeof(val));
}
void insert(char *s, int v)
{
int len = strlen(s), cur = 0;
for (int i=0; i<len; i++)
{
int idx = s[i] - 'a';
if (!ch[cur][idx])
{
ch[cur][idx] = ++next;
memset(ch[next], 0, sizeof(ch[next]));
}
cur = ch[cur][idx];
}
val[cur] += v;
}
void getFail()
{
queue<int>q;
f[0] = 0;
//初始化队列
for (int i=0; i<sigma_size; i++)
if ( ch[0][i] )
{
f[ ch[0][i] ] = 0;//第一个字母匹配失败只能指向0
last[ ch[0][i] ] = 0;//last后缀也只能指向0
q.push(ch[0][i]);
}
while (!q.empty())
{
int cur = q.front();
q.pop();
for (int i=0; i<sigma_size; i++)
{
int u = ch[cur][i];
//                        if (!u)//u结点下i位置没有字母
//                              continue;//继续下一个位置
if (!u)
{
ch[cur][i] = ch[ f[cur] ][i];
continue;
}
q.push(u);
int j = f[cur];//失败指针
while (j && !ch[j][i])//沿着失败指针走，ch[j][i]=0时没有此结点故继续
j = f[j];
f[u] = ch[j][i];//计算出结点u的失败指针
last[u] = val[ f[u] ] ? f[u] : last[ f[u] ];//f[u]处是否是单词结点
}
}
}
void getfail()
{
queue<int>q;
q.push(0);//根结点入队
while (!q.empty())
{
int cur = q.front();//取对头
q.pop();
for (int i=0; i<sigma_size; i++)
{
int u = ch[cur][i];
if (u)//u是否存在
{
if (cur==0)//如果父亲结点是根结点，失败指针和后缀链接指向0
f[u] = last[u] = 0;
else
{
int j = f[cur];//父亲的失败指针
while (j && !ch[j][i])//沿着父亲的失败指针走
j = f[j];
f[u] = ch[j][i];//找到u的失败指针
last[u] = val[ f[u] ] ? f[u] : last[ f[u] ];//f[u]处是否是单词结点
}
q.push( ch[cur][i] );
}
else
ch[cur][i] = ch[ f[cur] ][i];//不存在这一点，指向父亲结点的失败指针的第i个儿子
}
}
}
int find(char *t)
{
int len = strlen(t), j = 0, ans = 0, temp;
for (int i=0; i<len; i++)
{
int idx= t[i] - 'a';
//                  while (j && !ch[j][idx])//沿着失败指着走
//                        j = f[j];
j = ch[j][idx];//找到重新匹配的结点
temp = j;
while (temp && val[temp])//该结点是单词
{
ans += val[temp];//计数
val[temp] = 0;//标记已算过
temp = last[temp];//查找后缀串
}
}
return ans;
}
}ac;
int main()
{
int n, ncase;
scanf("%d", &ncase);
while (ncase--)
{
scanf("%d", &n);
ac.init();
for (int i=0; i<n; i++)
{
scanf("%s", s);
ac.insert(s, 1);
}
ac.getFail();
scanf("%s", t);
printf("%d\n", ac.find(t));
}
return 0;
}