bzoj 3172: [Tjoi2013]单词 AC自动机

3172: [Tjoi2013]单词

Time Limit: 20 Sec  Memory Limit: 256 MB

题目连接

http://www.lydsy.com/JudgeOnline/problem.php?id=3172

Description

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

HINT

题意

题解：

AC自动机，记录尾部的位置在哪儿

代码：

//qscqesze
#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <typeinfo>
#include <fstream>
#include <map>
#include <stack>
typedef long long ll;
using namespace std;
//freopen("D.in","r",stdin);
//freopen("D.out","w",stdout);
#define sspeed ios_base::sync_with_stdio(0);cin.tie(0)
#define test freopen("test.txt","r",stdin)
#define maxn 200001
#define mod 10007
#define eps 1e-9
int Num;
char CH[];
//const int inf=0x7fffffff;   //нчоч╢С
const int inf=0x3f3f3f3f;
inline ll read()
{
    ll x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
inline void P(int x)
{
    Num=;if(!x){putchar('');puts("");return;}
    while(x>)CH[++Num]=x%,x/=;
    while(Num)putchar(CH[Num--]+);
    puts("");
}
//**************************************************************************************

int n,pos[maxn];
int cnt;
int next[maxn][],sum[maxn],fail[maxn],q[maxn];
char s[maxn];
void insert(int &pos)
{
    int now=;
    scanf("%s",s);
    int n=strlen(s);
    for(int i=;i<n;i++)
    {
        if(!next[now][s[i]-'a'])
            next[now][s[i]-'a']=++cnt;
        now=next[now][s[i]-'a'];
        sum[now]++;
    }
    pos=now;
}
void buildfail()
{
    int head=,tail=;
    q[]=;fail[]=;
    while(head!=tail)
    {
        int now=q[head];
        head++;
        for(int i=;i<;i++)
        {
            int v=next[now][i];
            if(!v)
                continue;
            int k=fail[now];
            while(!next[k][i])
                k=fail[k];
            fail[v]=next[k][i];
            q[tail++]=v;
        }
    }
    for(int i=tail-;i>=;i--)
        sum[fail[q[i]]]+=sum[q[i]];
}

int main()
{
    //test;
    int n=read();
    cnt=;
    for(int i=;i<;i++)
        next[][i]=;
    for(int i=;i<=n;i++)
        insert(pos[i]);
    buildfail();
    for(int i=;i<=n;i++)
        printf("%d\n",sum[pos[i]]);
    return ;
}