3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 3852 Solved: 1864
[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⾃自动机
• ⽂文章在⾃自动机上跑⼀一遍
• 在每个状态上打标记
• 对每个单词,在fail树的⼦子树中求和即可
【代码】
//bzoj 3172 [Tjoi2013]单词
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
using namespace std;
const int mxn=1000010;
queue <int> q;
char s[mxn];
int w[mxn],wxl[mxn],n,num,tot;
struct node
{
int son[30];
int fail,cnt;
}a[mxn];
inline void clear(int x)
{
memset(a[x].son,0,sizeof a[x].son);
}
inline void trie(int u)
{
int x=0,len=strlen(s);
fo(i,0,len-1)
{
int t=s[i]-'a'+1;
if(!a[x].son[t])
{
num++;
clear(num);
a[x].son[t]=num;
}
x=a[x].son[t];
a[x].cnt++;
}
w[u]=x;
}
inline void build()
{
fo(i,1,26)
if(a[0].son[i])
q.push(a[0].son[i]);
while(!q.empty())
{
int x=q.front();
wxl[++tot]=x;
q.pop();
fo(i,1,26)
{
int y=a[x].son[i],fail=a[x].fail;
if(y)
a[y].fail=a[fail].son[i],q.push(y);
else
a[x].son[i]=a[fail].son[i];
}
}
}
int main()
{
scanf("%d",&n);
fo(i,1,n)
{
scanf("%s",s);
trie(i);
}
build();
for(int i=tot;i;i--)
a[a[wxl[i]].fail].cnt+=a[wxl[i]].cnt;
fo(i,1,n) printf("%d\n",a[w[i]].cnt);
return 0;
}