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