BZOJ2555 SubString 【后缀自动机 + LCT】
题目
懒得写背景了，给你一个字符串init，要求你支持两个操作
(1):在当前字符串的后面插入一个字符串
(2):询问字符串s在当前字符串中出现了几次？(作为连续子串)
你必须在线支持这些操作。
输入格式
第一行一个数Q表示操作个数
第二行一个字符串表示初始字符串init
接下来Q行，每行2个字符串Type,Str
Type是ADD的话表示在后面插入字符串。
Type是QUERY的话表示询问某字符串在当前字符串中出现了几次。
为了体现在线操作，你需要维护一个变量mask，初始值为0
读入串Str之后，使用这个过程将之解码成真正询问的串TrueStr。
询问的时候，对TrueStr询问后输出一行答案Result
然后mask = mask xor Result
插入的时候，将TrueStr插到当前字符串后面即可。
HINT:ADD和QUERY操作的字符串都需要解压
输出格式
输入样例
2
A
QUERY B
ADD BBABBBBAAB
输出样例
0
提示
40 % 的数据字符串最终长度 <= 20000，询问次数<= 1000，询问总长度<= 10000
100 % 的数据字符串最终长度 <= 600000，询问次数<= 10000,询问总长度<= 3000000
新加数据一组--2015.05.20
题解
大码题。。我就知道我会WA【到头来是解压错了= =，注意解压时的mask不是你维护的mask，你维护的mask值在解压前后不应改变】
由后缀自动机的性质，我们给SAM上的点赋值，在主链上的点权为1【即插入的不包括拓展的点】，其它点为0
权值为1的点在parent树上都是叶子节点，而一个节点所形成相同的串的数量就是该节点在parent树中的子树的叶子节点数
那这就好办了，我们用LCT维护parent树，每次加点就将ta到根节点加上ta的权值
由于树根始终不变而且树边有向，我们可以把LCT写得简单一点
【orz hzwer】
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define isr(u) (ch[fa[u]][1] == u)
#define isrt(u) (!fa[u] || (ch[fa[u]][0] != u && ch[fa[u]][1] != u))
#define ls ch[u][0]
#define rs ch[u][1]
using namespace std;
const int maxn = 1200005,maxm = 3000005,INF = 1000000000;
int fa[maxn],tag[maxn],val[maxn],ch[maxn][2];
int pre[maxn],son[maxn][26],step[maxn],cnt,last,n;
char s[maxm];
void add(int u,int v){if (u) val[u] += v,tag[u] += v;}
void pd(int u){
if (tag[u]) add(ls,tag[u]),add(rs,tag[u]),tag[u] = 0;
}
void push_down(int u){
if (!isrt(u)) push_down(fa[u]); pd(u);
}
void spin(int u){
int s = isr(u),f = fa[u];
fa[u] = fa[f]; if (!isrt(f)) ch[fa[f]][isr(f)] = u;
ch[f][s] = ch[u][s ^ 1]; if (ch[u][s ^ 1]) fa[ch[u][s ^ 1]] = f;
fa[f] = u; ch[u][s ^ 1] = f;
}
void splay(int u){
for (push_down(u); !isrt(u); spin(u))
if (!isrt(fa[u])) spin((isr(u) ^ isr(fa[u])) ? u : fa[u]);
}
void Access(int u){
for (int v = 0; u; u = fa[v = u]) splay(u),rs = v;
}
void Cut(int u){
Access(u); splay(u); add(ls,-val[u]);
fa[ls] = 0; ls = 0;
}
void Link(int u,int f){
fa[u] = f; Access(f); splay(f); add(f,val[u]);
}
void ins(int x){
int p = last,np = ++cnt;
last = np; val[np] = 1; step[np] = step[p] + 1;
while (p && !son[p][x]) son[p][x] = np,p = pre[p];
if (!p) pre[np] = 1,Link(np,1);
else {
int q = son[p][x];
if (step[q] == step[p] + 1) pre[np] = q,Link(np,q);
else {
int nq = ++cnt; step[nq] = step[p] + 1;
memcpy(son[nq],son[q],sizeof(son[q]));
pre[nq] = pre[q]; Link(nq,pre[q]);
pre[np] = nq; Link(np,nq);
Cut(q); pre[q] = nq; Link(q,nq);
while (son[p][x] == q) son[p][x] = nq,p = pre[p];
}
}
}
int MASK;
void Mask(int mask){
n = strlen(s);
for (int j = 0; j < n; j++){
mask = (mask * 131 + j) % n;
swap(s[j],s[mask]);
}
}
int solve(){
int u = 1; scanf("%s",s); Mask(MASK);
for (int i = 0; i < n; i++)
if (!(u = son[u][s[i] - 'A'])) return 0;
splay(u);
return val[u];
}
int main(){
int Q,ans; scanf("%d",&Q);
scanf("%s",s);
n = strlen(s); cnt = last = 1;
for (int i = 0; i < n; i++) ins(s[i] - 'A');
while (Q--){
scanf("%s",s);
if (s[0] == 'A'){
scanf("%s",s); Mask(MASK);
for (int i = 0; i < n; i++) ins(s[i] - 'A');
}
else printf("%d\n",ans = solve()),MASK ^= ans;
}
return 0;
}
