fail 树:一个点只有一个fail指针,所以把fail边倒过来就变成了一个树的结构。一个字符串出现在其他串中,当它是其他串一个后缀的前缀,例如abc出现在aaabcdd中,当它是abcdd这个后缀的前缀。所以需要算一下有多少fail直接或间接指向它最后一个字母c(fail树中 c子孙的大小),就可以算出abc是多少个后缀的前缀,也就可以算出它在整个文本中出现了多少次。
AC自动机建完后,用栈存fail深度,向前递推一遍可以求出。
第一天学,理解很有可能不对…………
#include<cstdio>
#include<iostream>
#include<stack>
#include<queue>
#include<cstring>
#define SZ 1000005
using namespace std;
struct Trie{
int ch[SZ][30],tot;
int d[SZ],val[SZ];
int lst[SZ],fail[SZ];
int id[SZ];
void insert(char *s,int num)
{
int p=0;
int l=strlen(s);
for(int i=0;i<l;i++)
{
int u=s[i]-'a'+1;
if(!ch[p][u]) ch[p][u]=++tot;
p=ch[p][u]; d[p]++;
}
val[p]++;
id[num]=p;
}
queue<int> q;
stack<int> S;
void build()
{
for(int i=1;i<=26;i++)
if(ch[0][i]) q.push(ch[0][i]);
while(!q.empty())
{
int r=q.front();q.pop();
S.push(r);
for(int i=1;i<=26;i++)
{
int u=ch[r][i];
if(!u){ch[r][i]=ch[fail[r]][i];continue;}
q.push(u);
int v=fail[r];
while(v&&!ch[v][i])v=fail[v];
fail[u]=ch[v][i];
lst[u]=val[fail[u]]?fail[u]:lst[fail[u]];
}
}
}
void up()
{
while(!S.empty())
{
int f=S.top();
S.pop();
d[fail[f]]+=d[f];
}
}
void print(int i)
{printf("%d\n",d[id[i]]);}
};
Trie AC;
char s[1000005];
int main()
{
int n;scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%s",s);
AC.insert(s,i);
}
AC.build();
AC.up();
for(int i=1;i<=n;i++)
AC.print(i);
}