[BZOJ2555]SubString(SAM+LCT)

2555: SubString

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 4077  Solved: 1236
[Submit][Status][Discuss]

Description

懒得写背景了，给你一个字符串init，要求你支持两个操作

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

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

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

Input

第一行一个数Q表示操作个数

第二行一个字符串表示初始字符串init

接下来Q行，每行2个字符串Type,Str

Type是ADD的话表示在后面插入字符串。

Type是QUERY的话表示询问某字符串在当前字符串中出现了几次。

为了体现在线操作，你需要维护一个变量mask，初始值为0 

    

读入串Str之后，使用这个过程将之解码成真正询问的串TrueStr。

询问的时候，对TrueStr询问后输出一行答案Result

然后mask=maskxorResult

插入的时候，将TrueStr插到当前字符串后面即可。

HINT:ADD和QUERY操作的字符串都需要解压

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

新加数据一组--2015.05.20

Output

Sample Input

2
A
QUERY B
ADD BBABBBBAAB

Sample Output

0

HINT

Source

Ctsc模拟赛By 洁妹

[Submit][Status][Discuss]

后缀自动机，每次查询的是一个点的Right。

发现每次改变fa[]实际上是在pre树上的一次Cut和Link操作，所以用LCT维护就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ls ch[x][0]
#define rs ch[x][1]
#define rep(i,l,r) for (int i=l; i<=r; i++)
using namespace std;

const int N=1200010;
int Q,len,lst=1,cnt=1,mask,f[N],sm[N],ch[N][2],tag[N],w[N],son[N][27],mx[N],fa[N];
char op[10],S[3000100];

bool isroot(int x){ return (!f[x]) || (ch[f[x]][0]!=x && ch[f[x]][1]!=x); }
void put(int x,int k){ w[x]+=k; tag[x]+=k; }
void push(int x){ if (tag[x]) put(ls,tag[x]),put(rs,tag[x]),tag[x]=0; }
void pd(int x){ if (!isroot(x)) pd(f[x]); push(x); }

void rot(int x){
    int y=f[x],z=f[y],w=ch[y][1]==x;
    if (!isroot(y)) ch[z][ch[z][1]==y]=x;
    f[x]=z; f[y]=x; f[ch[x][w^1]]=y;
    ch[y][w]=ch[x][w^1]; ch[x][w^1]=y;
}

void splay(int x){
    pd(x);
    while (!isroot(x)){
        int y=f[x],z=f[y];
        if (!isroot(y)) rot((ch[z][0]==y)^(ch[y][0]==x) ? x : y);
        rot(x);
    }
}

void access(int x){ for (int y=0; x; y=x,x=f[x]) splay(x),ch[x][1]=y; }
void link(int x,int y){ f[x]=y; access(y); splay(y); put(y,w[x]); }
void cut(int x){ access(x); splay(x); put(ls,-w[x]); f[ls]=0; ch[x][0]=0; }
int que(int x){ splay(x); return w[x]; }

void ext(int c){
    int p=lst,np=lst=++cnt; mx[np]=mx[p]+1; w[np]=1;
    while (p && !son[p][c]) son[p][c]=np,p=fa[p];
    if (!p) fa[np]=1,link(np,1);
    else{
        int q=son[p][c];
        if (mx[q]==mx[p]+1) fa[np]=q,link(np,q);
        else{
            int nq=++cnt; mx[nq]=mx[p]+1;
            while (p && son[p][c]==q) son[p][c]=nq,p=fa[p];
            memcpy(son[nq],son[q],sizeof(son[q]));
            fa[nq]=fa[q]; link(nq,fa[q]);
            cut(q); fa[np]=fa[q]=nq; link(np,nq); link(q,nq);
        }
    }
}

void get(int mask){
    scanf("%s",S); len=strlen(S);
    for (int i=0; i<len; i++) mask=(mask*131+i)%len,swap(S[i],S[mask]);
}

int main(){
    freopen("bzoj2555.in","r",stdin);
    freopen("bzoj2555.out","w",stdout);
    scanf("%d",&Q); scanf("%s",S); len=strlen(S);
    for (int i=0; i<len; i++) ext(S[i]-'A'+1);
    rep(i,1,Q){
        scanf("%s",op);
        if (op[0]=='A'){ get(mask); for (int j=0; j<len; j++) ext(S[j]-'A'+1); }
        else{
            get(mask); int x=1,flag=0;
            for (int j=0; j<len; j++){
                if (!son[x][S[j]-'A'+1]){ flag=1; puts("0"); break; }
                x=son[x][S[j]-'A'+1];
            }
            if (flag) continue;
            int t=que(x); mask^=t; printf("%d
",t);
        }
    }
    return 0;
}