hdu 3065 病毒侵袭持续中 ac自动机模板题

窝今天果断翻了SB，main函数中忘记写ac.bfs()这句话了，然后各种单步调试感觉怎么这个fail数组那么奇怪呢...很无语...后来不小心发现了果断AC，基本和上一道题类似，不过就是第几个字符串出现几次就对应的下标++，最后输出就好了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
#define N 2000000
int ans[1234];
char s[1234][52];
char v[3000000];
struct Trie
{
     int next[N][27],fail[N],mark[N];
    int root,L;
    int newnode()
    {
        memset(next[L],-1,sizeof(next[L]));
        mark[L]=0;
        fail[L++]=-1;
        return L-1;
    }
    void go()
    {
        L=0;
        root=newnode();
    }
    void init(char *v,int id)
    {
        int p=root;
        for(int i=0;v[i];i++)
        {
            int tep=v[i]-'A';
            if(next[p][tep]==-1) next[p][tep]=newnode();
            p=next[p][tep];
        }
        mark[p]=id;
    }
    void bfs()
    {
        queue<int>q;
        q.push(root);
        while(!q.empty())
        {
            int p=q.front();
            q.pop();
            for(int i=0;i<26;i++)
            {
                if(next[p][i]==-1)
                {
                    if(p==root) next[p][i]=root;
                    else next[p][i]=next[fail[p]][i];
                }
                else
                {
                    if(p==root) fail[next[p][i]]=root;
                    else fail[next[p][i]]=next[fail[p]][i];
                    q.push(next[p][i]);
                }
            }
        }
    }
    void finde(char *v)
    {
        int p=root,q;
        for(int i=0;v[i];i++)
        {
            if(v[i]<'A'||v[i]>'Z')
            {
                p=root;
                continue;
            }
            int tep=v[i]-'A';
            while(next[p][tep]==-1&&p!=root)
                p=fail[p];
            p=next[p][tep];
            if(p==-1) p=root;
            q=p;
            while(q!=root)
            {
                if(mark[q]!=0)
                    ans[mark[q]]++;
                q=fail[q];
            }
        }
    }
} ac;
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(ans,0,sizeof(ans));
        ac.go();
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s[i]);
            ac.init(s[i],i);
        }
        ac.bfs();
        scanf("%s",v);
        ac.finde(v);
        for(int i=1;i<=n;i++)
        {
            if(ans[i]==0)
                continue;
            printf("%s: %d\n",s[i],ans[i]);
        }
    }
    return 0;
}