[hdu3695]Computer Virus on Planet Pandora(AC自动机)

题意：正反匹配模式串。

解题关键：需要建立vis数组，不然会超时，因为匹配串会匹配很多次模式串，而我们只记录一次，从而第二次以后可以直接break。

复杂度可从$O(500W)$降到$O(25W)$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=26;
const int MAXN=250010;
bool vis[MAXN];
int ans;
struct Trie{//数组形式 
    int Next[MAXN][N],Fail[MAXN],End[MAXN],root,tot;//大小为所以匹配字符串的总和 
    int newnode(){//结构体内部用 
        for(int i=0;i<N;i++) Next[tot][i]=-1;
        End[tot++]=0;
        return tot-1;
    }
    void init(){
        tot=0;
        root=newnode();
    }
    void insert(char buf[]){
        int len=strlen(buf);
        int now=root;//now是temp指针 
        for(int i=0;i<len;i++){
            int k=buf[i]-'A';
            if(Next[now][k]==-1)  Next[now][k]=newnode();//next数组代表的是下一个字符索引 
            now=Next[now][k];
        }
        End[now]++;//end数组是当前字符串的个数.字典中可能有相同的单词，若只算一次，改为1. 
    }
    void build(){//构造fail指针，后缀是某些前缀 
        queue<int>que;
        Fail[root]=root;
        for(int i=0;i<N;i++){ 
            if(Next[root][i]==-1) Next[root][i]=root;
            else{
                Fail[Next[root][i]]=root;
                que.push(Next[root][i]);
            }
        } 
        while(!que.empty()){//bfs,会将所有的匹配子串都遍历到 
            int now=que.front();
            que.pop();
            for(int i=0;i<N;i++){
                if(Next[now][i]==-1) Next[now][i]=Next[Fail[now]][i];
                else{
                    Fail[Next[now][i]]=Next[Fail[now]][i];//fail指向最长的 
                    que.push(Next[now][i]);
                }
            }
        }
    }
    void query(char buf[]){
        int len=strlen(buf),now=root,res=0;
        for(int i=0;i<len;i++){
            now=Next[now][buf[i]-'A'];
            int temp=now;
            while(temp!=root){
                if(vis[temp]) break;
                 ans+=End[temp];
                 End[temp]=0;
                 vis[temp]=true;
                temp=Fail[temp];
            }
        }
    }
};

Trie ac;
char buf[5100010];
int n,m;
int input(){
    getchar();
    char ch;
    int slen=0;
    while((ch=getchar())!='
'){
        if(ch!='[') buf[slen++]=ch;
        else{
            int x;
            scanf("%d%c",&x,&ch);
            while(x--) buf[slen++]=ch;
            getchar();
        }
    }
    buf[slen]='';
    return slen;
}


int main(){
       int t;
       scanf("%d",&t);
       while(t--){
           memset(vis,0,sizeof vis);
           ans=0;
           ac.init();
           scanf("%d",&n);
           for(int i=0;i<n;i++){
               scanf("%s",buf);
               ac.insert(buf);
        }
        ac.build();
        int len=input();
        ac.query(buf);
        reverse(buf,buf+len);
        ac.query(buf);
        printf("%d
",ans);
    }
    return 0;
}