链接
题目描述
某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。
样例输入
3
a
aa
aaa
样例输出
6
3
1
思路
先用AC自动机
在串经过的节点都给它的sum值加上一
sum记录每个点到根所形成的串是多少个单词的前缀
然后用sum1表示答案,首先sum1显然等于sum
然后再想,nxt[u]一定在u前面,所以sum1[u] += sum1[nxt[u]]
最后把所有单词的结尾的sum1值输出就好了
代码
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n, tot, cnt;
int f[1000001], ch[10001][101], sum[1000001], q[100001], nxt[100001], r1[1000001];
char s[1000001];
void add()
{
int len = strlen(s);
int u = 1;
for(int i = 0; i < len; ++i)
{
if(!ch[u][s[i] - 'a']) ch[u][s[i] - 'a'] = ++tot;
u = ch[u][s[i] - 'a'];
sum[u]++;
}
f[++cnt] = u;
return ;
}
void bfs()
{
int q1 = 1, q2 = 0;
for(int i = 0; i <= 25; ++i)
if(ch[1][i]) {
q[++q2] = ch[1][i];
nxt[ch[1][i]] = 1;
}
for(q1 = 1; q1 <= q2; ++q1)
{
int u = q[q1];
for(int i = 0; i <= 25; ++i)
{
if(!ch[u][i]) ch[u][i] = ch[nxt[u]][i];
else
{
q[++q2] = ch[u][i];
nxt[ch[u][i]] = ch[nxt[u]][i];
}
}
}
for(int i = tot; i >= 1; --i)
r1[q[i]] = sum[q[i]];
for(int i = tot; i >= 1; --i)
r1[nxt[q[i]]] += r1[q[i]];
}
int main()
{
scanf("%d",&n);
tot = 1;
for(int i = 1; i <= n; ++i)
{
scanf("%s", s);
add();
}
bfs();
for(int i = 1; i <= n; ++i)
printf("%d\n",r1[f[i]]);
return 0;
}