hdu 3065 病毒侵袭持续中(AC自动机)

AC自动机第三题了，交上去1A真的爽，（题目没说多组坑了一发不能算，哼！）

其它oj不知道，hdu貌似对内存的统计是按使用过的内存来的，而不是按你申请的内存，然后静态字典树的话，只要动态使用数组就不会（其实就是不要一开始就整个初始化了）mle了。

然后当next[now][i]==-1的时候，直接让其等于next[fail[now]][i]这个优化真的不要太好，不仅时间上优化了，代码复杂度也是优化很多。

这道题唯一需要注意的地方就是，匹配的时候当匹配串出现不是26个英文大写字母的时候，直接让now=root，然后continue，因为病毒里，或者说字典树上没有 除这26个大写字母之外的字符。

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e6+5;
char str[maxn];
char a[1005][55];
int ans[1005];
int book[1005];
int k;
struct acho
{
    int nex[1000*50+10][27], cnt[1000*50+10], fail[1000*50+10];
    int L, root;
    int  newnode()
    {
        for(int i=0; i<26; i++)nex[L][i]=-1;
        cnt[L++]=0;
        return L-1;
    }
    void init()
    {
        L=0;
        root=newnode();
    }

    void insert(char *a, int id)
    {
        int now=root;
        for(int i=0; a[i]; i++)
        {
            if(nex[now][a[i]-'A']==-1)
            {
                nex[now][a[i]-'A']=newnode();
            }
            now=nex[now][a[i]-'A'];
        }
        cnt[now]=id;
        return;
    }

    queue<int>que;
    void build()
    {
        while(que.empty()==0)que.pop();
        int now=root;
        for(int i=0; i<26; i++)
        {
            if(nex[now][i]==-1)nex[now][i]=root;
            else 
            {
                fail[nex[now][i]]=root;
                que.push(nex[now][i]);
            } 
        }
        while(que.empty()==0)
        {
            now=que.front();
            que.pop();
            for(int i=0; i<26; i++)
            {
                if(nex[now][i]==-1)
                {
                    nex[now][i]=nex[fail[now]][i];
                }
                else 
                {
                    fail[nex[now][i]]=nex[fail[now]][i];
                    que.push(nex[now][i]);
                }
            }
        }
        return;
    }
    
    
    void match(char *a)
    {
        memset(book, 0, sizeof book);
        k=0;
        int now=root;
        for(int i=0; a[i]; i++)
        {
            if(a[i]<'A' || a[i]>'Z')
            {
                now=root;
                continue;
            }
            now=nex[now][a[i]-'A'];
            if(cnt[now]>0)
            {
                int v=now;
                while(v!=root)
                {
                    if(cnt[v]>0)
                    {
                        if(book[cnt[v]]==0)
                        {
                            ans[k++]=cnt[v];
                        }
                        book[cnt[v]]++;
                    }
                    v=fail[v];
                }
            }
        }
        return;
    }

}acho;



int main()
{
    int n, i, j;
    while(~scanf("%d", &n))
    {
    acho.init();
    for(i=1; i<=n; i++)
    {
        scanf("%s", a[i]);
        acho.insert(a[i], i);
    }
    acho.build();
    scanf("%s", str);
    acho.match(str);
    for(i=1; i<=n; i++)
    {
        if(book[i])printf("%s: %d\n", a[i], book[i]);
    }
    
    }
    
}