【BZOJ2555】SubString 后缀自动机+LCT

【BZOJ2555】SubString

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，询问总长度<= 10000
100 % 的数据字符串最终长度 <= 600000，询问次数<= 10000,询问总长度<= 3000000

题解：将串扔到SAM里，答案就是S的子树中结束节点的个数，由于强制在线只能用LCT维护，加入一个结束节点就将它到根的路径上的所有点权都+1。

discuss说不能用递归来下传标记？不明觉厉，写的非递归。

一开始写的太丑了，然后开始%黄学长的代码，%了一天一夜才搞出来QAQ

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=1200010;
int msk,n,m,ans,len;
struct LCT
{
int ch[maxn][2],s[maxn],tag[maxn],fa[maxn],q[maxn];
int isr(int x)  {return (ch[fa[x]][0]!=x)&&(ch[fa[x]][1]!=x);}
void pushdown(int x)
{
if(ch[x][0])    s[ch[x][0]]+=tag[x],tag[ch[x][0]]+=tag[x];
if(ch[x][1])    s[ch[x][1]]+=tag[x],tag[ch[x][1]]+=tag[x];
tag[x]=0;
}
void rotate(int x)
{
int y=fa[x],z=fa[y],d=(x==ch[y][1]);
if(!isr(y)) ch[z][y==ch[z][1]]=x;
fa[x]=z,fa[y]=x,ch[y][d]=ch[x][d^1];
if(ch[x][d^1])  fa[ch[x][d^1]]=y;
ch[x][d^1]=y;
}
void updata(int x)
{
q[0]=0;
while(1)
{
q[++q[0]]=x;
if(!isr(x)) x=fa[x];
else    break;
}
for(int i=q[0];i>=1;i--) pushdown(q[i]);
}
void splay(int x)
{
updata(x);
while(!isr(x))
{
int y=fa[x],z=fa[y];
if(!isr(y))
{
if((x==ch[y][0])^(y==ch[z][0])) rotate(x);
else    rotate(y);
}
rotate(x);
}
}
void access(int x)
{
int y=0;
while(x)    splay(x),ch[x][1]=y,y=x,x=fa[x];
}
void link(int x,int y)
{
fa[y]=x,access(x),splay(x),s[x]+=s[y],tag[x]+=s[y];
}
void cut(int x)
{
access(x),splay(x),s[ch[x][0]]-=s[x],tag[ch[x][0]]-=s[x];
fa[ch[x][0]]=0,ch[x][0]=0;
}
}T;
char str[maxn];
struct SAM
{
int pre[maxn],ch[maxn][26],dep[maxn],last,tot;
void add(int x)
{
int p=last,np=++tot;
last=np,dep[np]=dep[p]+1,T.s[np]=1;
for(;p&&!ch[p][x];p=pre[p]) ch[p][x]=np;
if(!p)  pre[np]=1,T.link(1,np);
else
{
int q=ch[p][x];
if(dep[q]==dep[p]+1)    pre[np]=q,T.link(q,np);
else
{
int nq=++tot;
dep[nq]=dep[p]+1,pre[nq]=pre[q],pre[q]=pre[np]=nq;
T.cut(q),T.link(pre[nq],nq),T.link(nq,q),T.link(nq,np);
memcpy(ch[nq],ch[q],sizeof(ch[q]));
for(;p&&ch[p][x]==q;p=pre[p])   ch[p][x]=nq;
}
}
}
int query()
{
int i,p=1;
for(i=p=1;i<=len&&p;i++) p=ch[p][str[i-1]-'A'];
if(!p)  return 0;
T.updata(p);
return T.s[p];
}
}S;
void decode()
{
scanf("%s",str),len=strlen(str);
int i,ms=msk;
for(i=0;i<len;i++)
{
ms=(ms*131+i)%len;
swap(str[i],str[ms]);
}
}
int main()
{
scanf("%d",&m);
S.last=S.tot=1;
int i,j;
scanf("%s",str),len=strlen(str);
for(i=1;i<=len;i++)  S.add(str[i-1]-'A');
for(i=1;i<=m;i++)
{
scanf("%s",str);
if(str[0]=='A')
{
decode();
for(j=1;j<=len;j++)  S.add(str[j-1]-'A');
}
else
{
decode(),ans=S.query(),msk^=ans;
printf("%d\n",ans);
}
}
return 0;
}