bzoj2555: SubString SAM+LCT

题目

bzoj2555

Description

懒得写背景了，给你一个字符串init，要求你支持两个操作
(1):在当前字符串的后面插入一个字符串
(2):询问字符串s在当前字符串中出现了几次？(作为连续子串)
你必须在线支持这些操作。

Input

第一行一个数Q表示操作个数
第二行一个字符串表示初始字符串init
接下来Q行，每行2个字符串Type,Str
Type是ADD的话表示在后面插入字符串。
Type是QUERY的话表示询问某字符串在当前字符串中出现了几次。
为了体现在线操作，你需要维护一个变量mask，初始值为0

读入串Str之后，使用这个过程将之解码成真正询问的串TrueStr。
询问的时候，对TrueStr询问后输出一行答案Result
然后mask = mask xor Result
插入的时候，将TrueStr插到当前字符串后面即可。

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

Output

Sample Input

2
A
QUERY B
ADD BBABBBBAAB

Sample Output

0

HINT

 40 % 的数据字符串最终长度 <= 20000，询问次数<= 1000，询问总长度<= 10000100 % 的数据字符串最终长度 <= 600000，询问次数<= 10000,询问总长度<= 3000000新加数据一组--2015.05.20

题意

给定一个初始字符串，支持2个操作。在后面添加一个字符串或询问一个字符串出现了多少次。 强制在线。

字符串最终长度 <= 600000，询问次数<= 10000,询问总长度<= 3000000。

题解

首先想到的是后缀自动机，每个节点维护一个size，添加一个字符时将它parent的size都加一。询问时沿着后缀自动机走，输出最后一个节点的size即可。

但暴力维护会TLE，于是只能上LCT了。

注意不仅要link新加进来的节点，在后缀自动机构建时parent改变后也要及时link和cut。还有就是复制节点q时所得到的nq的size是为0的（我最初设的是1，wa了好久，其实想想暴力维护时是怎么做的就明白了。），当link q与nq时会将nq的size累加上q的size。

下面贴代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;

#define maxn 1200100char s[3000010];
int n,len,ans,mask,last=1,tot=1;
struct LCT{
int son[maxn][2],size[maxn],f[maxn],tag[maxn];
#define root(x) (son[f[x]][0]!=x&&son[f[x]][1]!=x)

void push_down(int x){
if(tag[x]==0)return;
size[son[x][0]]+=tag[x];
tag[son[x][0]]+=tag[x];
size[son[x][1]]+=tag[x];
tag[son[x][1]]+=tag[x];
tag[x]=0;
}

void push_up(int x){
if(!root(x))push_up(f[x]);
push_down(x);
}

void rt(int x,int k){
int y=f[x],z=f[y];
son[y][k^1]=son[x][k];
f[son[x][k]]=y;
son[x][k]=y;
f[y]=x; f[x]=z;
if(son[z][0]==y)son[z][0]=x;
else if(son[z][1]==y)son[z][1]=x;
push_down(y);
push_down(x);
}

void splay(int x){
push_up(x);
while(!root(x)){
int y=f[x],z=f[y];
if(root(y)&&son[y][0]==x)rt(x,1);
else if(root(y)&&son[y][1]==x)rt(x,0);
else if(son[z][0]==y&&son[y][0]==x)rt(y,1),rt(x,1);
else if(son[z][0]==y&&son[y][1]==x)rt(x,0),rt(x,1);
else if(son[z][1]==y&&son[y][0]==x)rt(x,1),rt(x,0);
else rt(y,0),rt(x,0);
}
}

void access(int x){
for(int y=0;x;x=f[x]){
splay(x);
son[x][1]=y;
push_down(x);
y=x;
}
}

void link(int x,int y){
access(y);splay(y);f[x]=y;
size[y]+=size[x]; tag[y]+=size[x];
}

void cut(int x,int y){
access(y); splay(x); f[x]=0;
size[y]-=size[x]; tag[y]-=size[x];
}
}t;

struct SAM{
int go[maxn][26],pre[maxn],step[maxn];

void add(int x){
int p=last,np=++tot;
step[np]=step[p]+1;
for(;p!=0&&go[p][x]==0;p=pre[p])go[p][x]=np;
if(p==0)pre[np]=1;
else{
int q=go[p][x];
if(step[q]==step[p]+1)pre[np]=q;
else{
int nq=++tot;
memcpy(go[nq],go[q],sizeof(go[q]));
pre[nq]=pre[q];
t.link(nq,pre[q]);
step[nq]=step[p]+1;
t.cut(q,pre[q]);
pre[q]=pre[np]=nq;
t.link(q,nq);
for(;go[p][x]==q;p=pre[p])go[p][x]=nq;
}
}
last=np;
t.size[last]=1;
t.link(last,pre[last]);
}

int solve(char s[]){
int len=strlen(s),p=1;
for(int i=0;i<len;i++){
if(go[p][s[i]-'A']==0)return 0;
p=go[p][s[i]-'A'];
}
t.access(p);
t.splay(p);
return t.size[p];
}
}sam;

void trans(int mask){
scanf("%s",s);
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;
}
}

int main(){
scanf("%d",&n);
scanf("%s",s);
len=strlen(s);
for(int i=0;i<len;i++) sam.add(s[i]-'A');
for(int i=1;i<=n;i++){
scanf("%s",s);
if(s[0]=='A'){
trans(mask);
len=strlen(s);
for(int j=0;j<len;j++)
sam.add(s[j]-'A');
}else{
trans(mask);
ans=sam.solve(s);
printf("%d\n",ans);
mask^=ans;
}
}
return 0;
}