题意:给出\(n\)个单词,求每个单词在这\(n\)个单词中出现的次数。
\(fail\) 树:AC自动机中每个\(fail\)指针的反向边作为一条有向边,因为每个点只有\(1\)个\(fail\)指针,所以除了根节点每个点入度为\(1\),所以图是树形的。
那么对于这道题我们可以建出\(fail\)树,设\(siz_i\)表示有多少个单词经过i节点,所以每个节点的表示的字符串的出现次数应为该节点的\(fail\)树中子树的\(siz\)的和。
AC代码
#include<bits/stdc++.h>
#define N 201001
#define MAX 2001
#define re register
using namespace std;
typedef long long ll;
typedef double db;
inline void read(re ll &ret)
{
ret=0;re char c=getchar();re bool pd=false;
while(!isdigit(c)){pd|=c=='-';c=getchar();}
while(isdigit(c)){ret=(ret<<1)+(ret<<3)+(c&15);c=getchar();}
ret=pd?-ret:ret;
return;
}
ll n,trie[N][30],tot,nxt[N],num[N],ans,tt,siz[N],b[N],top;
char s[N][101],t[N*10];
ll f[N];
inline void insert(re ll pos)
{
re ll len(strlen(s[pos]+1)),p(0),c;
for(re int i(1);i<=len;++i)
{
c=s[pos][i]-'a';
if(!trie[p][c])
trie[p][c]=++tot;
p=trie[p][c];
siz[p]++;
}
f[pos]=p;
return;
}
inline void bfs()
{
queue<ll>q;
for(re int i=0;i<26;i++)
if(trie[0][i])
q.push(trie[0][i]);
re ll p;
while(!q.empty())
{
p=q.front();
b[++top]=p;
q.pop();
for(re int i(0);i<26;++i)
{
if(!trie[p][i])trie[p][i]=trie[nxt[p]][i];
else
{
q.push(trie[p][i]);
nxt[trie[p][i]]=trie[nxt[p]][i];
}
}
}
return;
}
inline void AC()
{
re ll len(strlen(t+1)),p(0),c,k;
for(re int i(1);i<=len;++i)
{
c=(t[i]-'a');
p=trie[p][c];
if(f[p])
{
for(re int j=p;j;j=nxt[j])
{
num[f[j]]++;
ans=max(ans,f[j]?num[f[j]]:0);
}
}
}
return;
}
main()
{
read(n);
if(!n)exit(0);
for(re int i(1);i<=n;i++)
{
scanf("%s",s[i]+1);
insert(i);
}
bfs();
for(re int i=tot;i>=0;i--)
siz[nxt[b[i]]]+=siz[b[i]];
for(re int i=1;i<=n;i++)
printf("%lld\n",siz[f[i]]);
exit(0);
}