BZOJ 2555 SubString LCT 後綴自動機

題目大意：給出一個初始字符串，要求支持，在末尾添加一段字符串，查詢一個串出現了幾次，強制在線。

構造後綴自動機，一個串出現次數等於代表這個串的狀態的right集合大小。

然而強制在線，right集合在添加之後會發生改變。
考慮添加一個字母時，只有產生的新狀態的所有parent的right集合大小會加1，於是用LCT維護一下就好了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 3000005
using namespace std;
namespace Link_Cut_Tree {
    struct Node {
        Node *ch[2],*pa;
        int dir() { return pa->ch[0]==this ? 0 : pa->ch[1]==this ? 1 : -1; }
        int val,add_mark;
        Node();
        void add(int);
        void pushdown();
        void* operator new(size_t) {
            static Node *mempool,*C;
            if(mempool==C) mempool=(C=new Node[1<<20])+(1<<20);
            return C++;
        }
    }*null=new Node();
    Node :: Node() {
        ch[0]=ch[1]=pa=null;
        val=add_mark=0;
    }
    void Node :: add(int x) {
        val+=x;
        add_mark+=x;
        return ;
    }
    void Node :: pushdown() {
        if(add_mark) {
            ch[0]->add(add_mark);
            ch[1]->add(add_mark);
            add_mark=0;
        }
        return ;
    }
    void Rotate(Node* o,int d) {
        Node* k=o->ch[d^1]; int d2;
        o->ch[d^1]=k->ch[d], k->ch[d]->pa=o;
        k->ch[d]=o;
        //o->maintain(), k->maintain();
        if(~(d2=o->dir())) o->pa->ch[d2]=k;
        k->pa=o->pa; o->pa=k;
        return ;
    }
    void To_pushdown(Node* o) {
        static Node* s[N];
        int top=0;
        while(~(o->dir())) s[++top]=o, o=o->pa;
        s[++top]=o;
        while(top) s[top--]->pushdown();
        return ;
    }
    void Splay(Node* o) {
        To_pushdown(o);
        int d;
        while(~(d=o->dir())) {
            if(o->pa->dir()==d) Rotate(o->pa->pa,d^1);
            Rotate(o->pa,d^1);
        }
        return ;
    }
    void Access(Node* o) {
        Node* p=null;
        while(o!=null) {
            Splay(o);
            o->ch[1]=p;
            p=o;
            o=o->pa;
        }
        return ;
    }
    void Cut(Node* x) {
        Access(x), Splay(x);
        x->ch[0]->pa=null;
        x->ch[0]=null;
        return ;
    }
    void Link(Node* x,Node* y) {
        Cut(x);
        x->pa=y;
        return ;
    }
}
#define LCT Link_Cut_Tree
namespace Suffix_Automaton {
    struct Node {
        Node *ch[26],*pa;
        int val;
        LCT::Node* o;
        Node(int _val=0):val(_val),pa(NULL) {
            memset(ch,0,sizeof ch);
            o=new LCT::Node();
        }
        void* operator new(size_t) {
            static Node *mempool,*C;
            if(mempool==C) mempool=(C=new Node[1<<20])+(1<<20);
            return C++;
        }
    }*root=new Node(0),*last=root;
    void Extend(char c) {
        int z=c-'A';
        Node *p=last,*np=new Node(p->val+1);
        last=np;
        while(p && !p->ch[z]) p->ch[z]=np, p=p->pa;
        if(!p) {
            np->pa=root;
            LCT::Link(np->o,root->o);
            LCT::Access(np->o), LCT::Splay(np->o);
            np->o->add(1);
            return ;
        }
        Node* q=p->ch[z];
        if(q->val==p->val+1) {
            np->pa=q;
            LCT::Link(np->o,q->o);
            LCT::Access(np->o), LCT::Splay(np->o);
            np->o->add(1);
            return ;
        }
        Node* nq=new Node(p->val+1);
        memcpy(nq->ch,q->ch,sizeof q->ch);
        LCT::Link(nq->o,q->pa->o);
        LCT::Link(q->o,nq->o);
        LCT::Link(np->o,nq->o);
        nq->pa=q->pa;
        q->pa=np->pa=nq;
        nq->o->val=q->o->val;
        while(p && p->ch[z]==q) p->ch[z]=nq, p=p->pa;
        LCT::Access(np->o), LCT::Splay(np->o);
        np->o->add(1);
        return ;
    }
    void Insert(char s[]) {
        int len=strlen(s);
        for(int i=0;i<len;i++) Extend(s[i]);
        return ;
    }
    int Query(char s[]) {
        Node* o=root;
        int len=strlen(s);
        for(int i=0;i<=len && o;i++)  {
            if(i==len) return LCT::To_pushdown(o->o), o->o->val;
            o=o->ch[s[i]-'A'];
        }
        return 0;
    }
}
#define SAM Suffix_Automaton
void decodeWithMask(char s[],int mask) {
    int len=strlen(s);
    for(int j=0;j<len;j++) {
        mask=(mask*131+j)%len;
        char t=s[j];
        s[j]=s[mask];
        s[mask]=t;
    }
    return ;
}
int main() {
    static char s[N];
    int T;
    scanf("%d%s",&T,s);
    SAM::Insert(s);
    int mask=0;
    while(T--) {
        char mode[10];
        scanf("%s%s",mode,s);
        decodeWithMask(s,mask);
        if(mode[0]=='Q') {
            int ans=SAM::Query(s);
            printf("%d\n",ans);
            mask^=ans;
        }
        else SAM::Insert(s);
    }
    return 0;
}