【bzoj3172】[Tjoi2013]单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 3034  Solved: 1447
[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

 

【题解】

很裸的AC自动机。

pos记录一下字符串结束的位置，每个结点初始权值都为1，结点的fail指针指向的点的权值要加上这个结点的权值。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
using namespace std;
#define MAXN 1000010
int n,cnt,v[MAXN],pos[MAXN],q[MAXN],fail[MAXN],tr[MAXN][27];
char ch[MAXN];
void insert(int &now)
{
    now=0;
    for(int i=1;i<=strlen(ch+1);i++)
    {
        if(!tr[now][ch[i]-'a']) tr[now][ch[i]-'a']=++cnt;
        now=tr[now][ch[i]-'a'];
        v[now]++;
    }
}
void build()
{
    int head=0,tail=0;
    for(int i=0;i<26;i++)  if(tr[0][i])  q[++tail]=tr[0][i];
    while(++head<=tail)
    {
        int x=q[head];
        for(int i=0;i<26;i++)
        {
            int y=tr[x][i];
            if(!y) continue;
            q[++tail]=y;
            int temp=fail[x];
            while(temp&&!tr[temp][i])  temp=fail[temp];
            fail[y]=tr[temp][i];
        }
    }
    for(int i=tail;i;i--)  v[fail[q[i]]]+=v[q[i]];
}
int main()
{
    //freopen("cin.in","r",stdin);
    //freopen("cout.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)  {scanf("%s",ch+1);  insert(pos[i]);}
    build();
    for(int i=1;i<=n;i++)  printf("%d\n",v[pos[i]]);
    return 0;
}