【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\n", 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