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

#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;
}