洛谷 P3966 [TJOI2013]单词（AC自动机， fail 树）

qq_38232157

于 2022-11-21 18:32:32 发布

阅读量286

点赞数

分类专栏：后缀数组文章标签： AC自动机

本文链接：https://blog.csdn.net/qq_38232157/article/details/127970079

版权

后缀数组专栏收录该内容

16 篇文章 0 订阅

订阅专栏

[TJOI2013]单词

题目描述

小张最近在忙毕设,所以一直在读论文。一篇论文是由许多单词组成但小张发现一个单词会在论文中出现很多次，他想知道每个单词分别在论文中出现了多少次。

输入格式

第一行一个整数 $N$ ，表示有 $N$ 个单词。

接下来 $N$ 行每行一个单词,每个单词都由小写字母 $a - z$ 组成。

输出格式

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

样例 #1

样例输入 #1

3
a
aa
aaa

样例输出 #1

6
3
1

提示

数据规模与约定

$30\%$ 的数据，单词总长度不超过 $10^3$ 。
$100\%$ 的数据， $\leq n \leq 200$ ，单词总长度不超过 $10^6$ 。

1、题目和 P5357 【模板】AC 自动机（二次加强版）一样，
不同的是，文本串有 n 个，也就是查询 query n次。

#include <bits/stdc++.h>
using namespace std;
const int N = 210, M = 1e6 + 10;
int ch[M][26], ne[M], idx;
string s[N];
int n;

int ans[M]; // ans[i]表示第i个字符串出现的次数
int mp[N];	// mp[x] 表示第x个字符串，对应的 AC自动机上的节点


int tot;
int ver[M], head[M], Next[M];

void add(int x, int y)
{
	ver[++tot] = y, Next[tot] = head[x], head[x] = tot;
}

void dfs(int x)
{
	for(int i = head[x]; i; i = Next[i])
	{
		int y = ver[i];
		dfs(y);
		ans[x] += ans[y];
	}
}

void insert(int k)
{
	int p = 0, len = s[k].size();
	for(int i = 0; i < len; ++i)
	{
		int j = s[k][i] - 'a';
		if(!ch[p][j])
			ch[p][j] = ++idx;
		p = ch[p][j];
	}
	mp[k] = p;
}

void build()
{
	queue<int> q;
	for(int i = 0; i < 26; ++i)
	{
		if(ch[0][i])
			q.push(ch[0][i]);
	}
	while(!q.empty())
	{
		int u = q.front(); q.pop();
		for(int i = 0; i < 26; ++i)
		{
			int v = ch[u][i];
			if(v)
			{
				ne[v] = ch[ne[u]][i], q.push(v);
			}else{
				ch[u][i] = ch[ne[u]][i];
			}
		}
	}
	for(int v = 1; v <= idx; ++v)
	{
		add(ne[v], v);
	}
}



void query(string s)
{
	int len = s.size();
	for(int k = 0, i = 0; k < len; ++k)
	{
		i = ch[i][s[k] - 'a'];
		ans[i]++;
	}
	return;
}

int main()
{
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i)
	{
		cin >> s[i];
		insert(i);
	}
	build();
	for(int i = 1; i <= n; ++i)
		query(s[i]);
	dfs(0);
	for(int i = 1; i <= n; ++i)
		printf("%d\n", ans[mp[i]]);
	return 0;
}