【BZOJ2555】SubString（后缀自动机，LCT）

题意：给你一个字符串init，要求你支持两个操作

(1):在当前字符串的后面插入一个字符串

(2):询问字符串s在当前字符串中出现了几次？(作为连续子串)

你必须在线支持这些操作。

长度 <= 600000，询问次数<= 10000,询问总长度<= 3000000

思路：因为有加边，删边，加点操作，需要动态维护SAM中每个right集合的大小，所以使用LCT维护

需要维护根节点=1号点到每个结点路径上的和，因为固定了1号点为根节点所以不需要makeroot和rev标记

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<ll,ll> Pll;
typedef vector<int> VI;
typedef vector<PII> VII;
typedef pair<ll,int>P;
#define N  1200010
#define M  210000
#define fi first
#define se second
#define MP make_pair
#define pi acos(-1)
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=(int)a;i<=(int)b;i++)
#define per(i,a,b) for(int i=(int)a;i>=(int)b;i--)
#define lowbit(x) x&(-x)
#define Rand (rand()*(1<<16)+rand())
#define id(x) ((x)<=B?(x):m-n/(x)+1)
#define ls p<<1
#define rs p<<1|1

const int MOD=1e9+7,inv2=(MOD+1)/2;
      double eps=1e-6;
      int INF=1<<29;
      ll inf=5e13;
      int dx[4]={-1,1,0,0};
      int dy[4]={0,0,-1,1};


int mask;
char s[3000010];
string chars;

int read()
{
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

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 t[N][2],w[N],fa[N],q[N],rev[N],tag[N],top;

    void add(int x,int y)
    {
        if(x)
        {
            w[x]+=y;
            tag[x]+=y;
        }
    }

    int isroot(int x)
    {
        return ((t[fa[x]][0]!=x)&&(t[fa[x]][1]!=x));
    }

    void pushdown(int x)
    {
        int l=t[x][0],r=t[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];
        int l=(t[y][1]==x),r=l^1;
        if(!isroot(y)) t[z][t[z][1]==y]=x;
        fa[t[x][r]]=y,fa[y]=x,fa[x]=z;
        t[y][l]=t[x][r],t[x][r]=y;
    }

    void splay(int x)
    {
        top=0;
        q[++top]=x;
        for(int i=x;!isroot(i);i=fa[i]) q[++top]=fa[i];
        while(top) pushdown(q[top--]);
        while(!isroot(x))
        {
            int y=fa[x],z=fa[y];
            if(!isroot(y))
            {
                if((t[y][0]==x)^(t[z][0]==y)) rotate(x);
                 else rotate(y);
            }
            rotate(x);
        }
    }

    void access(int x)
    {
        for(int k=0;x;k=x,x=fa[x])
        {
            splay(x);
            t[x][1]=k;
        }
    }

    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(t[x][0],-w[x]);
        fa[t[x][0]]=0;
        t[x][0]=0;
    }

}lct;

struct sam
{
    int cnt;
    int F[N],ch[N][26];
    int st[N],b[N],bl[N],c[N];
    int p,np,q,nq;

    sam()
    {
        cnt=np=1;
    }

    void extend(int x)
    {
        p=np; st[np=++cnt]=st[p]+1;
        while(p&&!ch[p][x])
        {
            ch[p][x]=np;
            p=F[p];
        }
        lct.w[np]=1;
        if(!p) F[np]=1,lct.link(np,1);
         else if(st[p]+1==st[q=ch[p][x]]) F[np]=q,lct.link(np,q);
          else
          {
              st[nq=++cnt]=st[p]+1;
              memcpy(ch[nq],ch[q],sizeof ch[q]);
              F[nq]=F[q];
              lct.link(nq,F[q]);
              F[np]=F[q]=nq;
              lct.cut(q);
              lct.link(q,nq);
              lct.link(np,nq);
              while(ch[p][x]==q)
              {
                    ch[p][x]=nq;
                    p=F[p];
              }
          }
    }

    void build()
    {
        scanf("%s",s);
        int n=strlen(s);
        rep(i,0,n-1) extend(s[i]-'A');
    }

    void add()
    {
        gets(mask);
        int n=chars.length();
        rep(i,0,n-1) extend(chars[i]-'A');
    }

    int query()
    {
        gets(mask);
        int p=1,n=chars.length();
        rep(i,0,n-1)
        {
            p=ch[p][chars[i]-'A'];
            if(!p) return 0;
        }
        lct.splay(p);
        return lct.w[p];
    }

}sam;


int main()
{
    int Q;
    scanf("%d",&Q);
    sam.build();
    while(Q--)
    {
        scanf("%s",s);
        if(s[0]=='A') sam.add();
         else
         {
            int ans=sam.query();
            printf("%d\n",ans);
            mask^=ans;
         }
    }
    return 0;
}