bzoj2555: SubString 后缀自动机+LCT

2555: SubString

Time Limit: 30 Sec   Memory Limit: 512 MB
Submit: 4189   Solved: 1284
[ 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]

考虑没有ADD操作，此时直接topo+dp然后跑一跑就行了。

我们dp的东西，在parent tree这棵内向树中，正是一个每个节点的子树大小。

但是这棵parent tree在增点nq时是会断了再重新接的。（其他时刻都是直接接上一个新点np）

于是考虑用LCT动态维护link与cut操作以及子树大小。由于没有makeroot，所以没有必要写update（但link和cut的时候依然要去更新）。每次修改操作直接处理根即可（打tag），然后再下放就好。

当然如果没发现这个省事的方法，你当然可以用我前面写的伪top-tree（大融合）的代码。

时间复杂度：操作都是logN的（alpha（字符集）也是近似于logN的）。于是大概O(NlogN)。 注意mask在transfer时是局部变量，并不真的乘131。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 600010
#define pa Fa[x]
#define lc ch[x][0]
#define rc ch[x][1]
using namespace std;
char ss[N];
int Q, n, mask = 0;
inline void getin(int mask) {
    scanf("%s", ss); n = strlen(ss);
    for(int i = 0; i < n; ++i) mask = (mask * 131 + i) % n, swap(ss[i], ss[mask]);
}
int ch[N<<1][2], Fa[N<<1], w[N<<1], add[N<<1];
inline bool wh(int x) {return ch[pa][1] == x;}
inline bool isr(int x) {return ch[pa][0] != x && ch[pa][1] != x;}
inline void plus(int x, int v) {add[x]+= v; w[x]+= v;}
inline void pushdown(int x) {
    if(add[x]) {
        if(lc) plus(lc, add[x]);
        if(rc) plus(rc, add[x]);
        add[x] = 0;
    }
}
inline void pd(int x) {if(!isr(x)) pd(pa); pushdown(x);}
inline void rotate(int x) {
    int y = Fa[x], z = Fa[y], c = wh(x);
    if(!isr(y)) ch[z][wh(y)] = x; Fa[x] = z;
    ch[y][c] = ch[x][c^1]; Fa[ch[x][c^1]] = y;
    ch[x][c^1] = y; Fa[y] = x;
}
inline void splay(int x) {
    pd(x);
    for(; !isr(x); rotate(x))
        if(!isr(pa)) rotate(wh(pa) == wh(x) ? pa : x);
}
inline void access(int x) {for(int y = 0; x; y = x, x = pa) splay(x), rc = y;}
inline void link(int x, int y) {
    Fa[x] = y;
    access(y); splay(y); plus(y, w[x]);
}
inline void cut(int x) {
    access(x); splay(x); plus(lc, -w[x]);
    Fa[lc] = 0; lc = 0;
}
int nxt[N<<1][26], fa[N<<1], mx[N<<1], cnt = 1, last = 1;
inline void extend(int c) {
    int np = ++cnt, p = last; mx[np] = mx[last] + 1; last = np;
    w[np] = 1;
    for(; p && !nxt[p][c]; p = fa[p]) nxt[p][c] = np;
    if(!p) fa[np] = 1, link(np, 1);
    else {
        int q = nxt[p][c];
        if(mx[q] == mx[p] + 1) fa[np] = q, link(np, q);
        else {
            int nq = ++cnt; mx[nq] = mx[p] + 1; fa[nq] = fa[q]; link(nq, fa[nq]); memcpy(nxt[nq], nxt[q], sizeof(nxt[nq]));
            for(; p && nxt[p][c] == q; p = fa[p]) nxt[p][c] = nq;
            cut(q); fa[q] = fa[np] = nq; link(q, nq); link(np, nq);
        }
    }
}
inline int query() {
    int u = 1;
    for(int i = 0; i < n; ++i) {
        int c = ss[i] - 'A';
        if(nxt[u][c]) u = nxt[u][c];
        else return 0;
    }
    pd(u);
    return w[u];
}
int main() {
    scanf("%d%s", &Q, ss); n = strlen(ss);
    for(int i = 0; i < n; ++i) extend(ss[i] - 'A');
    for(int i = 1; i <= Q; ++i) {
        char opt[10]; scanf("%s", opt); getin(mask);
        if(opt[0] == 'A') for(int j = 0; j < n; ++j) extend(ss[j] - 'A');
        else {
            int ans = query();
            printf("%d\n", ans);
            mask^= ans;
        }
    }
    return 0;
}