【BZOJ】2555: SubString（后缀自动机）

http://www.lydsy.com/JudgeOnline/problem.php?id=2555

学到了如何快速维护right值orz

（不过这仍然是暴力维护，可以卡到O(n)

首先我们在加一个新节点的时候直接遍历它的parent树就行啦啦！！！！这样直接就暴力维护了。。。我竟然没想到。。

然后在复制新节点的时候不要忘记将right也要赋值进去。。

然后就是裸题了

#include <cstdio>

#include <cstring>

#include <cmath>

#include <string>

#include <iostream>

#include <algorithm>

#include <queue>

#include <set>

#include <map>

using namespace std;

typedef long long ll;

#define rep(i, n) for(int i=0; i<(n); ++i)

#define for1(i,a,n) for(int i=(a);i<=(n);++i)

#define for2(i,a,n) for(int i=(a);i<(n);++i)

#define for3(i,a,n) for(int i=(a);i>=(n);--i)

#define for4(i,a,n) for(int i=(a);i>(n);--i)

#define CC(i,a) memset(i,a,sizeof(i))

#define read(a) a=getint()

#define print(a) printf("%d", a)

#define dbg(x) cout << (#x) << " = " << (x) << endl

#define error(x) (!(x)?puts("error"):0)

#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)

inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<'0'||c>'9'; c=getchar()) if(c=='-') k=-1; for(; c>='0'&&c<='9'; c=getchar()) r=r*10+c-'0'; return k*r; }

struct sam {

	static const int N=1500004;

	int l[N], c[N][26], f[N], root, last, cnt, r[N];

	sam() { cnt=0; root=last=++cnt; }

	void add(int x) {

		int now=last, a=++cnt; last=a;

		l[a]=l[now]+1;

		for(; now && !c[now][x]; now=f[now]) c[now][x]=a;

		if(!now) f[a]=root;

		else {

			int q=c[now][x];

			if(l[q]==l[now]+1) f[a]=q;

			else {

				int b=++cnt;

				memcpy(c[b], c[q], sizeof c[q]);

				l[b]=l[now]+1;

				r[b]=r[q];

				f[b]=f[q];

				f[q]=f[a]=b;

				for(; now && c[now][x]==q; now=f[now]) c[now][x]=b;

			}

		}

		while(a) ++r[a], a=f[a];

	}

	int find(char *s) {

		int now=root, n=strlen(s);

		rep(i, n) if(!(now=c[now][s[i]-'A'])) return 0;

		return r[now];

	}

}a;

void fix(char *s, int last) {

	int len=strlen(s);

	rep(i, len) {

		last=(last*131+i)%len;

		swap(s[i], s[last]);

	}

}

const int N=3000005;

char s[N], w[100];

int main() {

	int n=getint(), len, last=0;

	scanf("%s", s);

	len=strlen(s);

	rep(i, len) a.add(s[i]-'A');

	for1(i, 1, n) {

		scanf("%s%s", w, s);

		len=strlen(s);

		fix(s, last);

		if(w[0]=='Q') { int ans=a.find(s); printf("%d\n", ans); last^=ans; }

		else rep(i, len) a.add(s[i]-'A');

	}

	return 0;

}

Description

懒得写背景了，给你一个字符串init，要求你支持两个操作

    (1):在当前字符串的后面插入一个字符串

    (2):询问字符串s在当前字符串中出现了几次？(作为连续子串)

    你必须在线支持这些操作。

Input

第一行一个数Q表示操作个数

    第二行一个字符串表示初始字符串init

    接下来Q行，每行2个字符串Type,Str

    Type是ADD的话表示在后面插入字符串。

    Type是QUERY的话表示询问某字符串在当前字符串中出现了几次。

    为了体现在线操作，你需要维护一个变量mask，初始值为0

【BZOJ】2555: SubString（后缀自动机）-LMLPHP
    读入串Str之后，使用这个过程将之解码成真正询问的串TrueStr。
    询问的时候，对TrueStr询问后输出一行答案Result
    然后mask = mask xor Result
    插入的时候，将TrueStr插到当前字符串后面即可。

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

Output

Sample Input

QUERY B

ADD BBABBBBAAB

Sample Output

HINT

40 % 的数据字符串最终长度 <= 20000，询问次数<= 1000，询问总长度<= 10000

100 % 的数据字符串最终长度 <= 600000，询问次数<= 10000,询问总长度<= 3000000

Source

Ctsc模拟赛By 洁妹