[bzoj3172][Tjoi2013]单词

来源：互联网发布：classic动作数据mmd 编辑：程序博客网时间：2024/05/29 13:50

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

aaa
Sample Output

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;}

0 0