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;
}