BZOJ 2555: SubString 动态树+后缀自动机

首先，这是一道代码题，利用动态树维护后缀自动机的parent树，如果不是强制在线，可能会好写一点。

对于每个询问，找到被询问节点所在的right集合，集合大小即为答案。

注意，这里的LCT为有向的，写起来有些不一样。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 3050000;
int n,m,mask,T;
char s[maxn];
string chars;
void gets(int mask)
{
    scanf("%s",s);
    chars=s;
    for (int j=0;j<chars.length();j++) 
    {
        mask=(mask*131+j)%chars.length();
        char t=chars[j];
        chars[j]=chars[mask];
        chars[mask]=t;
    }
}
struct lct{
    int w[maxn],ch[maxn][2],fa[maxn],tag[maxn],q[maxn];
    bool rev[maxn];
    bool isroot(int x){
        return ch[fa[x]][0]!=x&&ch[fa[x]][1]!=x;
    }
    void add(int x,int y)
    {
        if(x)
        {
            w[x]+=y;tag[x]+=y;
        }
    }
    void pushdown(int x)
    {
        int l=ch[x][0],r=ch[x][1];
        if(tag[x])
        {
            add(l,tag[x]),add(r,tag[x]);
            tag[x]=0;
        }
    }
    void rotate(int x)
    {
        int y=fa[x],z=fa[y],l=ch[y][1]==x,r=l^1;
        if(!isroot(y)) ch[z][ch[z][1]==y]=x;
        fa[x]=z;fa[y]=x;fa[ch[x][r]]=y;
        ch[y][l]=ch[x][r],ch[x][r]=y;
    }
    void splay(int x){
        int top=0;q[++top]=x;
        for(int i=x;!isroot(i);i=fa[i])
            q[++top]=fa[i];//cout<<q[i]<<endl;
        for(int i=top;i;i--) pushdown(q[i]);
        while(!isroot(x))
        {
            int y=fa[x],z=fa[y];
            if(!isroot(y))
            {
                if(ch[y][0]==x^ch[z][0]==y) rotate(x);
                else rotate(y);
            }
            rotate(x);
        }
    }
    void access(int x)
    {
        int t=0;
        while(x)
        {
            splay(x);ch[x][1]=t;t=x;x=fa[x];
        }
    }
    void link(int x,int y){
        fa[x]=y;access(y);splay(y);add(y,w[x]);
    }
    void cut(int x){
        access(x);splay(x);add(ch[x][0],-w[x]);
        fa[ch[x][0]]=0;ch[x][0]=0;
    }
}t;
struct sam{
    int last,cnt;
    int a[maxn][27],fa[maxn],mx[maxn];
    sam(){
        last=++cnt;
    }   
    void insert(int c)
    {
        int p=last,np=last=++cnt;t.w[np]=1;mx[np]=mx[p]+1;
        while(!a[p][c]&&p) a[p][c]=np,p=fa[p];
        if(!p) fa[np]=1,t.link(np,1);
        else
        {
            int q=a[p][c];
            if(mx[p]+1==mx[q]) fa[np]=q,t.link(np,q);
            else 
            {
                int nq=++cnt;mx[nq]=mx[p]+1;
                memcpy(a[nq],a[q],sizeof(a[q]));
                fa[nq]=fa[q];
                t.link(nq,fa[q]);
                fa[np]=fa[q]=nq;
                t.cut(q);t.link(np,nq);t.link(q,nq);
                while(a[p][c]==q) a[p][c]=nq,p=fa[p];
            }
        }
    }
    void build(){
        scanf("%s",s);
        int l=strlen(s);
        for(int i=0;i<l;i++) insert(s[i]-'A');
    }
    void add(){
        gets(mask);
        int l=chars.length();
        for(int i=0;i<l;i++) insert(chars[i]-'A');
    }
    int query(){
        gets(mask);
        int l=chars.length(),p=1;
        for(int i=0;i<l;i++)
            if(!(p=a[p][chars[i]-'A'])) return 0;
        t.splay(p);
        return t.w[p];
    }
} sam;
char op[5];
int main()
{
    scanf("%d",&T);
    sam.build();
    while(T--)
    {
        scanf("%s",op);
        if(op[0]=='A') sam.add();
        else 
        {
            int ans=sam.query();
            printf("%d\n",ans);
            mask^=ans;
        }
    }
    return 0;
}