HDOJ 3065 病毒侵袭持续中 （AC自动机）

题意：给定n个单词（均为大写字母）和一个文本（均为可见字符），求每个单词在文本中出现的次数。单词数不超过1000，长度不超过50，文本长度不超过2000000

分析：AC自动机基础题。

一个小的优化：由于单词均为大写字母，所以建字典树时第二维大小可以只开26，这样可以节约时间和空间，在扫描文本时（均为可见字符），碰到非大写字母时，直接将指针指向根结点，继续扫描下一个字符即可。

#include <stdio.h>
#include <string.h>
#include <queue>
using namespace std;
#define LEN 51
#define N 1001
char str[N][LEN];
char t[2000001];
int n,node;
int next[N*LEN][26];
int fail[N*LEN];
int id[N*LEN];
int ans[N];
void init()
{
    node=1;
    memset(next[0],0,sizeof(next[0]));
    memset(ans,0,sizeof(ans));
}
void add(int cur,int k)
{
    memset(next[node],0,sizeof(next[node]));
    id[node]=0;
    next[cur][k]=node++;
}
void insert(int k,char s[])
{
    int i,j,cur=0;
    for(i=0;s[i];i++)
    {
        j=s[i]-'A';
        if(!next[cur][j])   add(cur,j);
        cur=next[cur][j];
    }
    id[cur]=k;
}
void CalFail()
{
    int i,k,cur,nxt,tmp;
    queue<int>q;

    fail[0]=0;
    q.push(0);

    while(!q.empty())
    {
        cur=q.front(),q.pop();
        for(i=0;i<26;i++)
        {
            nxt=next[cur][i];
            if(!nxt)    continue;

            if(!cur)    fail[nxt]=0;
            else
            {
                for(tmp=fail[cur];tmp && !next[tmp][i];tmp=fail[tmp]);
                fail[nxt]=next[tmp][i];
            }
            q.push(nxt);
        }
    }
}
void match(char s[])
{
    int i,k,cur=0,tmp;
    for(i=0;s[i];i++)
    {
        k=s[i]-'A';
        if(k<0 || k>=26)    {cur=0;continue;}
        for(;cur && !next[cur][k];cur=fail[cur]);
        cur=next[cur][k];

        for(tmp=cur;tmp;tmp=fail[tmp])
        {
            ans[id[tmp]]++;
        }
    }
}
int main()
{
    int i;
    while(~scanf("%d",&n))
    {
        init();
        for(i=1;i<=n;i++)
        {
            scanf("%s",str[i]);
            insert(i,str[i]);
        }
        scanf("%s",t);
        CalFail();
        match(t);
        for(i=1;i<=n;i++)
        {
            if(ans[i])   printf("%s: %d\n",str[i],ans[i]);
        }
    }
    return 0;
}