hdu_4787_GRE Words Revenge(在线AC自动机)

题目链接：hdu_4787_GRE Words Revenge

题意：

总共有n个操作，2种操作。每行读入一个字符串。

1.如果字符串以+开头，此为单词（即模式串，不考虑重复）

2.如果字符串以？开头，此为文章（即文本串，查询在此之前的单词在文本串中出现的次数）

题解：

强制在线的AC自动机

贴个大牛的详细题解http://blog.csdn.net/no__stop/article/details/16823479

这样的带合并操作的AC自动机用第二种建树的方式比较方便

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std;

const int AC_N=1e5+7,tyn=2,M=5e6+10;//数量乘串长,类型数
struct AC_automation{
    int tr[AC_N][tyn],cnt[AC_N],Q[AC_N],fail[AC_N],tot;
    void nw(){cnt[++tot]=0;memset(tr[tot],-1,sizeof(tr[tot]));}
    void init(){tot=-1,fail[0]=-1,nw();}
    void insert(char *s,int x=0){
        for(int i=0,w;s[i];x=tr[x][w],i++)
            if(tr[x][w=s[i]-'0']==-1)nw(),tr[x][w]=tot;
        cnt[x]=1;//串尾标记
    }
    void build(int head=1,int tail=0){
        F(i,0,1)if(~tr[0][i])Q[++tail]=tr[0][i],fail[tr[0][i]]=0;
        while(head<=tail){
            for(int i=0,x=Q[head++],p=-1;i<tyn;i++)if(~tr[x][i]){
                for(p=fail[x],fail[tr[x][i]]=0;~p;p=fail[p])
                    if(~tr[p][i]){fail[tr[x][i]]=tr[p][i];break;}
                Q[++tail]=tr[x][i];
            }
        }
    }
    int ask(char *s,int ans=0){
        for(int w,i=0,x=0,j;s[i];i++){
            while(tr[x][w=s[i]-'0']==-1&&x)x=fail[x];
            x=tr[x][w],x=(~x)?x:0,j=x;
            while(j){if(cnt[j])ans++;j=fail[j];}
        }
        return ans;
    }
    
}a,b;

char s[M],tps[M];
int t,n,an,sqr=2000,bcnt;

void dfs(int r1,int r2)//将b中以r2为根结点的树合并到a中以r1为根结点的树中
{
    F(i,0,1)if(~b.tr[r2][i])
    {
        if(a.tr[r1][i]==-1)a.nw(),a.tr[r1][i]=a.tot;
        a.cnt[a.tr[r1][i]]|=b.cnt[b.tr[r2][i]];
        dfs(a.tr[r1][i],b.tr[r2][i]);
    }
}

void join(){dfs(0,0),b.init(),a.build();}//将b合并到a中

void decrypt()
{
    int len=strlen(s);
    int k=an%(len-1),ed=0;
    F(i,1,len-1)tps[i]=s[i];
    F(i,k+1,len-1)s[++ed]=tps[i];
    F(i,1,k)s[++ed]=tps[i];
}

int main()
{
    scanf("%d",&t);
    F(ic,1,t)
    {
        printf("Case #%d:
",ic);
        scanf("%d",&n);
        a.init(),b.init(),an=bcnt=0;
        F(i,1,n)
        {
            scanf("%s",s),decrypt();
            if(s[0]=='+')
            {
                b.insert(s+1),b.build(),bcnt++;
                if(bcnt>sqr)join();
            }else printf("%d
",(an=a.ask(s+1)+b.ask(s+1)));
        }
    }
    return 0;
}