传送门

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;
}
posted on 2016-02-14 20:08  geng4512  阅读(299)  评论(0编辑  收藏  举报