[bzoj3172][Tjoi2013]单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 3845 Solved: 1859
[Submit][Status][Discuss]
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

利用fail指针的特点

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
const int N = 1000000 + 5;
const int P = 30;
int n;
int sz=0,ans[N],num[N];
int query[N],cnt,fail[N],ch[N][P];
char s[N];
void insert( int &yu ){
    int now = 0, len = strlen(s);
    for( int i = 0; i < len; i++ ){
        int x = s[i] - 'a';
        if( !ch[now][x] ) ch[now][x] = ++sz;
        now = ch[now][x];
        ans[now]++;
    }
    yu = now;
}
queue<int> q;
void failure(){
    while( !q.empty() ){ q.pop(); }
    for( int i = 0; i < 26; i++ )
        if( ch[0][i] ) q.push(ch[0][i]);
    while( !q.empty() ){
        int now = q.front(); q.pop();
        query[++cnt] = now;
        for( int i = 0; i < 26; i++ ){
            if( !ch[now][i] ){
                ch[now][i] = ch[fail[now]][i]; continue;
            }
            fail[ch[now][i]] = ch[fail[now]][i];
            q.push(ch[now][i]);
        }
    }
}
int main(){
    scanf("%d", &n);
    for( int i = 1; i <= n; i++ ){
        scanf("%s", s);
        insert(num[i]);
    }
    failure();
    for( int i = cnt; i >= 1; i-- )
        ans[fail[query[i]]] += ans[query[i]];
    for( int i = 1; i <= n; i++ )
        printf("%d\n",ans[num[i]]);
    return 0;
}