BZOJ3172单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec Memory Limit: 512 MB

Submit: 2536 Solved: 1185

Description

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

Input

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

Output

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

Sample Input

3

a

aa

aaa

Sample Output

6

3

1

借鉴的黄学长的模板。。

建出fail树之后。。对每个子串的每个节点打标记。。

求出每个子串的终点标记个数和。。

附上本蒟蒻的代码：

#include<cstdio>
#include<cstring>
using namespace std;
#define MAXN 1000001
int n,pos[MAXN];
struct kx
{
int cnt,next[MAXN][26],sum[MAXN],fail[MAXN],q[MAXN];
char ch[MAXN];
kx()
{
cnt=1;
for (int i=0;i<26;i++) next[0][i]=1;
}
void insert(int &pos)
{
int now=1;
scanf("%s",&ch);
int n=strlen(ch);
for (int i=0;i<n;i++)
{
if (!next[now][ch[i]-'a']) next[now][ch[i]-'a']=++cnt;
now=next[now][ch[i]-'a'];
sum[now]++;
}
pos=now;
}
void build()
{
int head=0,tail=1;
q[0]=1,fail[1]=0;
while (head!=tail)
{
int now=q[head];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]];
}
}kx;

int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
kx.insert(pos[i]);
kx.build();
for (int i=1;i<=n;i++)
printf("%d\n",kx.sum[pos[i]]);
return 0;
}