BZOJ3172 TJOI2013 单词

传送门

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

哎，这不是多字符串的问题吗？我们首先就想到了AC自动机！
fail指针的意义是什么呢？就是一个后缀链接，而后缀是覆盖了所有的子串的，所以我们可以用一次树DP，就统计出了每一个单词出现的次数。

/**************************************************************Problem: 3172User: geng4512Language: C++Result: AcceptedTime:256 msMemory:237132 kb
****************************************************************/#include<cstdio>
#define MAXN 2000000
#define MAXC 26
int a[MAXN][MAXC], pre[MAXN], sz = 1, q[MAXN], cnt[MAXN], ed[MAXN], n;
char s[MAXN];
inline int Insert(char *s) {int c, p = 1;for(int i = 0; s[i]; ++ i) {c = s[i] - 'a';if(!a[p][c]) a[p][c] = ++ sz;p = a[p][c]; ++ cnt[p];}return p;
}
void Build() {int l = 1, r = 0;for(int i = 0; i < MAXC; ++ i)if(!a[1][i]) a[1][i] = 1;else {pre[a[1][i]] = 1;q[++ r] = a[1][i];}while(l <= r) {int u = q[l ++];for(int i = 0; i < MAXC; ++ i)if(!a[u][i]) a[u][i] = a[pre[u]][i];else {pre[a[u][i]] = a[pre[u]][i];q[++ r] = a[u][i];}}for(int i = r; i; -- i) cnt[pre[q[i]]] += cnt[q[i]];
}
int main() {scanf("%d", &n);for(int i = 1; i <= n; ++ i) {scanf("%s", s);ed[i] = Insert(s);}Build();for(int i = 1; i <= n; ++ i) printf("%d\n", cnt[ed[i]]);return 0;
}