题意:
一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。
思路:
ac自动机的本质如果一个串是另外一个串的子串,那么这个串是另外一个串的前缀的后缀,即另外一个串通过不停地fail指针可以移动
到这个串的末尾知道了这个理论,那么其实fail指针就是一棵树,求这个串总共出现多少次->这个节点的子树的大小(通过循环传递可以直接求出)
#include<bits/stdc++.h>
using namespace std;
char s[1150010];
int ed[1000010],q[1000010];
struct Trie{
int sz,val[1000010],next[1000010][26],fail[1000010],root;
int newnode(){
memset(next[sz],-1,sizeof(next[sz]));
val[sz]=0,sz++;
return sz-1;
}
void init(){
sz=0;
root=newnode();
}
void insert(int st,int ed,int &pos){
int now=root;
for(int i=st;i<ed;i++){
if(next[now][s[i]-'a']==-1)
next[now][s[i]-'a']=newnode();
now=next[now][s[i]-'a'];
val[now]++;
}
pos=now;
}
void build(){
int head=0,tail=0;
fail[root]=root;
for(int i=0;i<26;i++)
if(next[root][i]!=-1)
fail[next[root][i]]=root,q[++tail]=next[root][i];
while(head<tail){
int now=q[++head];
for(int i=0;i<26;i++){
if(next[now][i]!=-1){
int k=fail[now];
while(k!=root&&next[k][i]==-1) k=fail[k];
if(next[k][i]!=-1)
k=next[k][i];
fail[next[now][i]]=k;
q[++tail]=next[now][i];
}
}
}
for(int i=tail;i>=1;i--)
val[fail[q[i]]]+=val[q[i]];
}
};
Trie ac;
int len[1000];
int main(){
ac.init();
int n;
scanf("%d",&n);
len[0]=0;
for(int i=1;i<=n;i++){
scanf("%s",s+len[i-1]);
len[i]=strlen(s);
ac.insert(len[i-1],len[i],ed[i]);
}
ac.build();
for(int i=1;i<=n;i++)
printf("%d\n",ac.val[ed[i]]);
return 0;
}