FOJ2005

题意：给你一些字符串和一些文章，问这篇文章和它倒着读组成的文章一共出现多少个给定的字符串。

AC自动机

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

const int kind = 26;
struct node{
    node *fail;       
    node *next[kind]; 
    int count;        
    node(){           
        fail=NULL;
        count=0;
        memset(next,NULL,sizeof(next));
    }
}*q[250005];          
char keyword[1005];     
char str[5000005],str1[5000005];    
int head,tail;        
int value(int p,int q)
{
    int i,ans=0,w=1;
    for (i=q;i>=p;i--)
    {
        ans+=(str1[i]-'0')*w;
        w*=10;
    }

    return ans;
}
void insert(char *str,node *root){
    node *p=root;
    int i=0,index;
    while(str[i]){
        index=str[i]-'A';
        if(p->next[index]==NULL) p->next[index]=new node();
        p=p->next[index];
        i++;
    }
    p->count++;
}
void build_ac_automation(node *root){
    int i;
    root->fail=NULL;
    q[head++]=root;
    while(head!=tail){
        node *temp=q[tail++];
        node *p=NULL;
        for(i=0;i<26;i++){
            if(temp->next[i]!=NULL){
                if(temp==root) temp->next[i]->fail=root;
                else{
                    p=temp->fail;
                    while(p!=NULL){
                        if(p->next[i]!=NULL){
                            temp->next[i]->fail=p->next[i];
                            break;
                        }
                        p=p->fail;
                    }
                    if(p==NULL) temp->next[i]->fail=root;
                }
                q[head++]=temp->next[i];
            }
        }
    }
}
int query(node *root){
    int i=0,cnt=0,index,len=strlen(str);
    node *p=root;
    while(str[i]){
        index=str[i]-'A';
        while(p->next[index]==NULL && p!=root) p=p->fail;
        p=p->next[index];
        p=(p==NULL)?root:p;
        node *temp=p;
        while(temp!=root && temp->count!=-1){
            cnt+=temp->count;
            temp->count=-1;
            temp=temp->fail;
        }
        i++;
    }
    return cnt;
}
int main(){
    int n,t,i,j,l,k;
    scanf("%d",&t);
    while(t--){
        head=tail=0;
        node *root=new node();
        scanf("%d",&n);
        getchar();
        while(n--){
            gets(keyword);
            insert(keyword,root);
        }
        build_ac_automation(root);
        gets(str1);
        l=strlen(str1);
        j=0;
        for (i=0;i<l;)
        {
            if (str1[i]!='[') str[j++]=str1[i++];
            else
            {
                for (k=i+1;str1[k]!=']';k++);
                int v=value(i+1,k-2);
                for (int k1=1;k1<=v;k1++) str[j++]=str1[k-1];
                i=k+1;
            }
        }
        str[j]='\0';
        int h=query(root);
        char chh;
        l=strlen(str);
        for (i=0;i<=(l-1)/2;i++)
        {
            chh=str[l-i-1];
            str[l-i-1]=str[i];
            str[i]=chh;
        }
        h+=query(root);
        printf("%d\n",h);
    }
    return 0;
}