【bzoj 3172】【codevs 2542】[Tjoi2013]单词（AC自动机）

3172: [Tjoi2013]单词

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 3024  Solved: 1441
[Submit][Status][Discuss]

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

HINT

Source

[Submit][Status][Discuss]

【题解】【AC自动机】

【类似于“病毒持续侵袭中”，也要记录单词的出现位置和结束位置，但不能在最后查找，因为单词出现的次数是从它出现的位置开始算起，即所有单词合成的这篇文章中，处于当前单词之前可能还合成了当前单词，但不计入最后结果】

#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
vector<int>loc,order;
int n,nxt[1000010][30],fail[1000010],sum[1000010],tot;
bool vis[1000010];
inline void clean()
{
	memset(sum,0,sizeof(sum));
	memset(fail,0,sizeof(fail));
	memset(nxt,0,sizeof(nxt));
	memset(vis,false,sizeof(vis));
	tot=0;
	loc.clear(); order.clear(); 
	return;
 } 
inline void insert(char c[])
{
	int len=strlen(c);
	int now=0,i;
	for (i=0;i<len;++i)
	 {
	 	int x=c[i]-'a';
	 	if (!nxt[now][x]) nxt[now][x]=++tot;
	 	sum[nxt[now][x]]++; now=nxt[now][x];
	 }
	vis[now]=true;//标记以当前点为结尾的单词出现过 
	loc.push_back(now);//标记结尾 
	return;
}
inline void buildfail()
{
	queue<int>que;
    int i,j;
	for(i=0;i<26;++i) 
	  if (nxt[0][i]) que.push(nxt[0][i]);
	while (!que.empty())
	 {
	 	int now=que.front(); que.pop();
	 	order.push_back(now);//把当前出现的单词存起来 
	 	for (i=0;i<26;++i)
	 	 {
	 	 	if (!nxt[now][i])
	 	 	  {nxt[now][i]=nxt[fail[now]][i]; continue;}
	 	 	int x=nxt[now][i];
	 	 	fail[x]=nxt[fail[now]][i];//fail里存到当前字母为止的单词的最长后缀 
	 	    que.push(x);//将更新出来的单词存入队列（存的是序号） 
	    } 
	  } 
	return;
}
inline void find()
{
  for (int i=order.size()-1;i>=0;--i)
   sum[fail[order[i]]]+=sum[order[i]];
  for (int i=0;i<n;++i)
   printf("%d\n",sum[loc[i]]);
  return;
}

int main()
{
	int i,j;
	clean();
	scanf("%d",&n);
	for (i=0;i<n;++i)
	 {
	    char s1[1000010];
	    scanf("%s",s1);
	 	insert(s1);
	 }
	buildfail();
	find();
	return 0;
}