[BZOJ3172]单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec  Memory Limit: 512 MB

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

 

题解：

这道题相对来说比较简单……

我们可以建好AC自动机，然后像后缀数组一样，把每一个串用‘&’连起来。

像这样：“aaa&aa&a”

然后拿这个大串去AC自动机里面统计次数就好了。

代码见下：

 #include<cstdio>
 #include<cstring>
 #include<vector>
 using namespace std;
 const int L=;
 char text[*L],s[][L];
 int n,p=-,hd=,tl;
 long long ans[];
 struct node
 {
     vector<int>val;node *last,*f,*ch[];
     node(){memset(ch,,sizeof(ch));val.clear();}
     void* operator new(size_t);
 }*root=new node(),*c,*num,*q[L*];
 void* node::operator new(size_t)
 {
     if(c==num)
     {
         c=new node::node[<<];
         memset(c,,sizeof(node)*(<<));
         num=c+(<<);
     }
     return c++;
 }
 inline void count(node* pos)
 {
     while(pos)
     {
         for(int i=;i<pos->val.size();i++)
             ans[pos->val[i]]++;
         pos=pos->last;
     }
 }
 inline void get_fail()
 {
     root->f=NULL;root->last=NULL;q[++tl]=root;
     while(hd<=tl)
     {
         node *rt=q[hd++];
         for(int i=;i<;i++)
         {
             if(rt->ch[i])
             {
                 q[++tl]=rt->ch[i];
                 node *u=rt->f;
                 while(u&&!u->ch[i])u=u->f;
                 rt->ch[i]->f=(u)?u->ch[i]:root;
                 rt->ch[i]->last=rt->ch[i]->f->val.empty()?rt->ch[i]->f->last:rt->ch[i]->f;
             }
         }
     }
 }
 inline void match()
 {
     node *rt=root;
     for(int i=;i<p;i++)
     {
         if(text[i]=='&')rt=root;
         else
         {
             int d=text[i]-'a';
             while(rt&&!rt->ch[d])rt=rt->f;
             rt=rt?rt->ch[d]:root;
             count(rt);
         }
     }
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;i++)
     {
         scanf("%s",s[i]);int m=strlen(s[i]);
         node* rt=root;
         for(int j=;j<m;j++)
         {
             text[++p]=s[i][j];
             if(!rt->ch[s[i][j]-'a'])
                 rt->ch[s[i][j]-'a']=new node();
             rt=rt->ch[s[i][j]-'a'];
         }
         rt->val.push_back(i);text[++p]='&';
     }
     get_fail();match();
     for(int i=;i<=n;i++)printf("%lld\n",ans[i]);
 }

BZOJ3172