【AC自动机】单词频率

链接

单词频率

题目描述

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

样例输入

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