BZOJ 3172

3172: [Tjoi2013]单词

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

AC自动机

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 1000005
using namespace std;
int pos[N],n;
struct acm
{
    int cnt;
    int next[N][26],q[N],sum[N],fail[N];
    char ch[N];
    acm()
    {
        cnt=1;
        for(int i=0;i<26;i++)next[0][i]=1;
    }
    void insert(int &pos)
    {
        int now=1;
        scanf("%s",ch);
        int len=strlen(ch);
        for(int i=0;i<len;i++)
        {
            if(!next[now][ch[i]-'a'])next[now][ch[i]-'a']=++cnt;
            now=next[now][ch[i]-'a'];
            sum[now]++;
        }
        pos=now;
    }
    void buildfail()
    {
        int head=0,tail=1;
        q[0]=1;fail[1]=0;
        while(head!=tail)
        {
            int now=q[head++];
            for(int i=0;i<26;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-1;i>=0;i--)
            sum[fail[q[i]]]+=sum[q[i]];
    }
}trie;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)trie.insert(pos[i]);
    trie.buildfail();
    for(int i=1;i<=n;i++)printf("%d
",trie.sum[pos[i]]);
//    system("pause");
    return 0;
}
        

这题bzoj上我都能排进rank前50？？