BZOJ 3172 [Tjoi2013] 单词 [AC自动机模板]

Description

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

Input

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

Output

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

Sample Input

3

a

aa

aaa

Sample Output

6

3

1

HINT

Source
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

AC自动机~

#include<cstdio>
#include<cstring>

int n,c[1000001][28],fail[1000001],q[1000001],w[1000001],cnt,num[1000001],head,tail;
char s[1000001];

struct ac{
void add(int &u)
{
int x=0,len=strlen(s);
for(int i=0;i<len;i++)
{
if(!c[x][s[i]-'a']) c[x][s[i]-'a']=++cnt;
num[x=c[x][s[i]-'a']]++;u=x;
}
}
void getfail()
{
tail=1;fail[0]=-1;
while(head!=tail)
{
int k=q[++head];
for(int i=0;i<26;i++)
if(c[k][i]) q[++tail]=c[k][i],fail[c[k][i]]=k ? c[fail[k]][i]:0;
else c[k][i]=k ? c[fail[k]][i]:0;
}
}
void cal()
{
for(int i=tail;i;i--) num[fail[q[i]]]+=num[q[i]];
for(int i=1;i<=n;i++) printf("%d\n",num[w[i]]);
}
}ac;

int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%s",s),ac.add(w[i]);
ac.getfail();ac.cal();
return 0;
}