HDU3595 Computer Virus on Planet Pandora （AC自动机）

HDU3595 Computer Virus on Planet Pandora

原题地址：http://acm.hdu.edu.cn/showproblem.php?pid=3695

题意：
多组数据。
每组数据给n个模式串和一个主串(长度<=q)，输出有多少个模式串在主串中出现过（倒着也算）。主串给的形式是压缩格式。
数据范围
T<= 10， 0 < n <= 250，0 < q <= 5,100,000
题解：
裸AC自动机。
倒着怎么办？直接再把主串倒着跑一边就行。跑过的把isword设成零。
不用指针之后真是相当舒爽。
代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N=5100010;
const int M=500010; 
queue<int> Q;
int T,n,ch[M][26];
char s[N],t[N],str[1095];
int last[N],fail[N],tail=0,root=0,isword[N],ans=0;
void init()
{
    memset(ch,0,sizeof(ch));
    memset(last,0,sizeof(last));
    memset(fail,0,sizeof(fail));
    memset(isword,0,sizeof(isword));
    tail=0; ans=0;
}
void insert()
{
    int len=strlen(str);
    int p=0;
    for(int i=0;i<len;i++)
    {
        int c=str[i]-'A';
        if(!ch[p][c]) ch[p][c]=++tail;
        p=ch[p][c];
    }
    isword[p]++;
}
void getfail()
{
    while(!Q.empty()) Q.pop();
    for(int i=0;i<26;i++)
    {
        if(ch[0][i])
        {
            fail[ch[0][i]]=last[ch[0][i]]=0;
            Q.push(ch[0][i]);
        }
    }
    while(!Q.empty())
    {
        int top=Q.front(); Q.pop();
        for(int i=0;i<26;i++)
        {
            if(!ch[top][i]) 
            {ch[top][i]=ch[fail[top]][i]; continue;}

            int now=ch[top][i];
            fail[now]=ch[fail[top]][i];
            if(isword[fail[now]]) last[now]=fail[now]; else last[now]=last[fail[now]];  
            Q.push(now);        
        }
    }
}
int upcomp()
{
    int len=strlen(t);
    int now=0; int i=0;
    while(i<len)
    {
        if(t[i]=='[')
        {
            int num=0;
            i++;
            while(t[i]>='0'&&t[i]<='9')
            num=num*10+t[i++]-'0'; 
            for(int j=1;j<=num;j++)
            s[now++]=t[i];          
            i++;
        }
        else if(t[i]>='A'&&t[i]<='Z')
        s[now++]=t[i];
        i++;
    }
    return now;
}
void tadd(int x)
{
    while(x)
    {
        ans+=isword[x]; isword[x]=0; 
        int tmp=x; x=last[x]; last[tmp]=0;
    }
}
void solve(int len)
{
    int p=0;
    for(int i=0;i<len;i++)
    {
        int c=s[i]-'A';
        while(!ch[p][c]&&p) p=fail[p];
        p=ch[p][c];
        if(p)
        {
            if(isword[p])
            {tadd(p); last[p]=0;}
            else
            {tadd(last[p]); last[p]=0;}
        }
    }
    p=0;
    for(int i=len-1;i>=0;i--)
    {
        int c=s[i]-'A';
        while(!ch[p][c]&&p) p=fail[p];
        p=ch[p][c];
        if(p)
        {
            if(isword[p])
            {tadd(p); last[p]=0;}
            else
            {tadd(last[p]); last[p]=0;}
        }
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%s",str);
            insert();
        }
        getfail();
        scanf("%s",t);
        int len=upcomp();
        solve(len);
        printf("%d\n",ans);
    }
    return 0;
}