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: 3172 User: geng4512 Language: C++ Result: Accepted Time:256 ms Memory: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;
}