[BZOJ]2555 Substring 后缀自动机&LCT

2555: SubString

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 3115  Solved: 925
[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 = 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

新加数据一组--2015.05.20

   

Source

Ctsc模拟赛By 洁妹

[Submit][Status][Discuss]

HOME Back

　　hh代码题...要求动态插入当然是用sam啦... 由sam里的parent的性质可以得到， 一个串的出现次数就是在sam里面跑出来到的节点的在parent树里的子树siz， 也就是到达点的 |right|. 由于要动态插入并且sam中会有砍掉par边， 加入par边的操作... 于是就想到用lct来维护啦. 这道题是有向图都不用makeroot... 虽然说是代码题但还是挺好写的， 也就100行.

　　需要注意的细节: 在访问一个节点的w一定要记得先把标记传下来， 要不然就会得到错误的答案。 话说样例查询不存在的串, 加了串后又不查询是有多水? 不过还是帮我查出re233. 一发AC一颗赛艇.

#include<bits/stdc++.h>using namespace std;const int maxn = 2e6 + 5;string chars;char ss[maxn];int top, root, tot, mask, last, Q, ans;int dep[maxn], w[maxn], tag[maxn], c[maxn][2], fa[maxn], s[maxn], a[maxn][26], par[maxn];inline void decr(int mask) {scanf("%s", ss);chars = ss;for (int j = 0; j < chars.length(); ++ j) {mask = (mask * 131 + j) % chars.length();char t = chars[j];chars[j] = chars[mask];chars[mask] = t;}}inline bool isroot(const int &x) {return c[fa[x]][0] != x && c[fa[x]][1] != x;}inline void add(int x, int val) {if (x) w[x] += val, tag[x] += val;}inline void pushdown(int x) {if (tag[x]) {add(c[x][0], tag[x]), add(c[x][1], tag[x]);tag[x] = 0;}}inline void rotate(int x) {int y = fa[x], z = fa[y];int l = (c[y][1] == x), r = l ^ 1;if (!isroot(y)) c[z][c[z][1] == y] = x;fa[x] = z, fa[y] = x, fa[c[x][r]] = y;c[y][l] = c[x][r], c[x][r] = y;}inline void splay(int x) {top = 0;s[++ top] = x;for (int i = x; !isroot(i); i = fa[i]) s[++ top] = fa[i];for (int i = top; i; -- i) pushdown(s[i]);for (int f; !isroot(x); rotate(x))if (!isroot(f = fa[x]))rotate((c[fa[f]][0] == f ^ c[f][0] == x) ? x : f);}inline void access(int x) {for (int t = 0; x; x = fa[x])splay(x), c[x][1] = t, t = x;}inline void link(int x, int y) {fa[x] = y, access(y), splay(y), add(y, w[x]);}inline void cut(int x) {access(x), splay(x), add(c[x][0], -w[x]);fa[c[x][0]] = 0, c[x][0] = 0;}inline int query() {decr(mask);int len = chars.length(), p = 1;for (int j = 0; j < len; ++ j)if (!(p = a[p][chars[j] - 'A'])) return 0;splay(p);return w[p];}inline void insert(int c) {int p = last, np = last = ++ tot;w[np] = 1, dep[np] = dep[p] + 1;while (p && !a[p][c]) a[p][c] = np, p = par[p];if (!p) par[np] = root, link(np, root);else {int q = a[p][c];if (dep[q] == dep[p] + 1) par[np] = q, link(np, q);else {int nq = ++ tot;dep[nq] = dep[p] + 1;par[nq] = par[q], link(nq, par[q]);memcpy(a[nq], a[q], sizeof(a[q]));par[np] = par[q] = nq;cut(q), link(q, nq), link(np, nq);while (a[p][c] == q) a[p][c] = nq, p = par[p];}}}int main() {root = last = ++ tot;scanf("%d", &Q);scanf("%s", ss);for (register int i = 0; ss[i]; ++ i)insert(ss[i] - 'A');for (register int i = 0; i < Q; ++ i) {scanf("%s", ss);if (ss[0] == 'A') {decr(mask);int len = chars.length();for (int j = 0; j < len; ++ j)insert(chars[j] - 'A');} else {ans = query(), mask ^= ans;printf("%d\n", ans);}}}