题目大意
有 n 个单词,然后问你每个单词在这些单词中出现了多少次。
思路
显然的AC自动机,这时我们考虑每一个节点到根的字符串是多少字符的前缀即可。
code:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
using namespace std;
int trie[1000006][26],que[1000006],s[1000006][2],tot=1,tot2,ed[1000006],nxt[1000006],n,m;
string x[210];
void insert(string x)
{
int p=1;
for (int i=0;i<x.size();i++)
{
if (!trie[p][x[i]-'a']) trie[p][x[i]-'a']=++tot;
p=trie[p][x[i]-'a'];
s[p][0]++;
}
ed[++tot2]=p;
return;
}
void bfs()
{
queue<int> o;
for (int i=0;i<26;i++) trie[0][i]=1;
o.push(1);
nxt[1]=0;
int r=0;
que[r++]=1;
while (o.size())
{
int u=o.front();
o.pop();
for (int i=0;i<26;i++)
{
if (!trie[u][i]) trie[u][i]=trie[nxt[u]][i];
else
{
o.push(trie[u][i]);
que[r++]=trie[u][i];
nxt[trie[u][i]]=trie[nxt[u]][i];
}
}
}
for (int i=r-1;i>=0;i--) s[que[i]][1]=s[que[i]][0];
for (int i=r-1;i>=0;i--) s[nxt[que[i]]][1]+=s[que[i]][1];
return;
}
int ACQuantAsk(string x)
{
int ans=0,u=1,l=x.size();
for (int i=0;i<l;i++)
{
int k=trie[u][x[i]-'a'];
while (k>1&&ed[k]!=-1)
{
ans+=ed[k];
ed[k]=-1;
k=nxt[k];
}
u=trie[u][x[i]-'a'];
}
return ans;
}
string xxx;
int main()
{
for (int i=0;i<26;i++) trie[0][i]=1,trie[1][i]=0;
tot=1;
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
cin>>x[i];
insert(x[i]);
}
bfs();
for (int i=1;i<=n;i++) cout<<s[ed[i]][1]<<endl;
return 0;
}