【BZOJ3172】单词（AC自动机）

题面

Description

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

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

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

Sample Input

3

a

aa

aaa

Sample Output

6

3

1

题解

yyb因为调这道题而死亡

在重复一下题目的意思把。。。

看不见文章对不读，，

那是因为文章就是所有单词组成的

然后你就可以yy所有单词中间有一个空格之类的东西

很明显的AC自动机，

然后，我们

每次把每个单词带进去匹配一下

暴跳fail指针

美滋滋的收获90分

0分是空间玩炸了。。。

暴跳fail指针是可以被卡炸的。。。

所以，我们就不暴跳了呀

每次要跳的时候就在这个点这里打一个标记

所有标记打完之后

我们就从底下往上一层层跳（记一下bfs序倒着跳）

每次就只跳一层，然后标记丢到上面去

这样就可以一起跳啦

然后美滋滋的 AC啦

对了，

这题还有一点

就是会有重复的单词。。。

所以要记录一下每个单词和哪个单词是一样的（不是并查集）

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define MAX 1100000

struct Node

{

	int vis[26];

	int fail,ff;

	int id,sum;

}t[MAX];

int tot,lid[500];

int ans[500];

int n,q[MAX],tp;

char ch[MAX],ss[MAX];

void Insert(int id,char *s)

{

	int gg=strlen(s);

	int now=0;

	for(int i=0;i<gg;++i)

	{

		if(!t[now].vis[s[i]-'a'])

			t[now].vis[s[i]-'a']=++tot;

		t[t[now].vis[s[i]-'a']].ff=now;

		now=t[now].vis[s[i]-'a'];

	}

	if(!t[now].id)t[now].id=id,lid[id]=id;

	else lid[id]=t[now].id;

}

queue<int> Q;

void GetFail()

{

	for(int i=0;i<26;++i)

		if(t[0].vis[i])

			Q.push(q[++tp]=t[0].vis[i]);

	while(!Q.empty())

	{

		int u=Q.front();Q.pop();

		for(int i=0;i<26;++i)

		{

			if(t[u].vis[i])

				t[t[u].vis[i]].fail=t[t[u].fail].vis[i],Q.push(q[++tp]=t[u].vis[i]);

			else

				t[u].vis[i]=t[t[u].fail].vis[i];

		}

	}

}

int l=0;

int main()

{

	scanf("%d",&n);

	for(int i=1;i<=n;++i)

	{

		scanf("%s",ch);

		Insert(i,ch);

		int len=strlen(ch);

		for(int j=0;j<len;++j)

			ss[l++]=ch[j];

		ss[l++]='#';

	}

	GetFail();

	int now=0;

	for(int j=0;j<l;++j)

	{

		if(ss[j]=='#')now=0;

		else now=t[now].vis[ss[j]-'a'];

		t[now].sum++;

	}

	for(int i=tp;i;i--)

	{

		ans[t[q[i]].id]+=t[q[i]].sum;

		t[t[q[i]].fail].sum+=t[q[i]].sum;

	}

	for(int i=1;i<=n;++i)printf("%d\n",ans[lid[i]]);

	return 0;

}