【BZOJ】3172: [Tjoi2013]单词（后缀自动机）

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

随便搞个sam就行了。（其实一开始看到数据n<=200, 单词长度不超过1e6，然后感觉。。。200*1e6?。完全不可做。。可是后来看到别人都只开了1e6的数组，那么放心了。。题目没说清楚。

在做着题时发现个问题囧。就是并不是自动机上每个状态都是right初始化为1后再进行更新，而是每一个原串转移到的状态为1。这点不用多说吧。。。

#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=2*(1e6+215);
	int l[N], c[N][27], 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; return; }
		int q=c[now][x];
		if(l[q]==l[now]+1) { f[a]=q; return; }
		int b=++cnt;
		memcpy(c[b], c[q], sizeof c[q]);
		l[b]=l[now]+1;
		f[b]=f[q];
		f[q]=f[a]=b;
		for(; now && c[now][x]==q; now=f[now]) c[now][x]=b;
	}
	void build(int *s, int len) {
		rep(i, len) add(s[i]);
		int now=root;
		rep(i, len) now=c[now][s[i]], ++r[now];
		static int b[N], t[N];
		for1(i, 1, cnt) t[l[i]]++;
		for1(i, 1, len) t[i]+=t[i-1];
		for1(i, 1, cnt) b[t[l[i]]--]=i;
		for3(i, cnt, 1) r[f[b[i]]]+=r[b[i]];
	}
	int find(int *s, int len) {
		int now=root;
		rep(i, len) if(!(now=c[now][s[i]])) return 0;
		return r[now];
	}
}a;
const int N=1e6+215;
int s[N], w[N];
int main() {
	int n=getint(), len=0;
	for1(i, 1, n) {
		char c=getchar(); while(c<'a'||c>'z') c=getchar();
		while(c>='a'&&c<='z') s[len++]=c-'a', c=getchar();
		s[len++]=26;
	}
	a.build(s, len);
	int i=0;
	while(i<len) {
		int cnt=0, c=s[i++];
		while(i<len && c==26) c=s[i++];
		while(i<len && c!=26) w[cnt++]=c, c=s[i++];
		printf("%d
", a.find(w, cnt));
	}
	return 0;
}

　　

---

Description

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

HINT

 

Source