题解 [TJOI2013]单词

题意：给出$n$个单词，求每个单词在这$n$个单词中出现的次数。

$fail$ 树：AC自动机中每个$fail$指针的反向边作为一条有向边，因为每个点只有$1$个$fail$指针，所以除了根节点每个点入度为$1$，所以图是树形的。

那么对于这道题我们可以建出$fail$树，设$siz_i$表示有多少个单词经过i节点，所以每个节点的表示的字符串的出现次数应为该节点的$fail$树中子树的$siz$的和。

AC代码

Copy
#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);
}