近乎模板,只要暴力建树,再把每个字符串丢进去匹配就可以了。
我一开始写了非拓扑的代码,结果最后一个点T了。
不得不改成了拓扑。
另外,这题需要保存字符串信息,为了避免MLE,我们把所有东西全都压成一个大串。
代码:
#include<bits/stdc++.h>
using namespace std;
int n,cnt=1,occ[210],S[210];
char s[1000001];
struct AC_automaton{
int ch[26],fail,in,tms;
vector<int>fin;
}t[1001000];
inline void ins(int id){
int x=1;
for(register int i=S[id-1];i<S[id];i++){
if(!t[x].ch[s[i]-'a'])t[x].ch[s[i]-'a']=++cnt;
x=t[x].ch[s[i]-'a'];
}
t[x].fin.push_back(id);
}
queue<int>q;
inline void build(){
for(register int i=0;i<26;i++){
if(t[1].ch[i])t[t[1].ch[i]].fail=1,q.push(t[1].ch[i]),t[1].in++;
else t[1].ch[i]=1;
}
while(!q.empty()){
register int x=q.front();q.pop();
for(register int i=0;i<26;i++){
if(t[x].ch[i])t[t[x].ch[i]].fail=t[t[x].fail].ch[i],q.push(t[x].ch[i]),t[t[t[x].fail].ch[i]].in++;
else t[x].ch[i]=t[t[x].fail].ch[i];
}
}
}
inline void merge(int id){
int x=1;
for(register int i=S[id-1];i<S[id];i++){
x=t[x].ch[s[i]-'a'];
t[x].tms++;
}
}
inline void topo(){
for(int i=1;i<=cnt;i++)if(!t[i].in)q.push(i);
while(!q.empty()){
int x=q.front();q.pop();
t[t[x].fail].in--;
t[t[x].fail].tms+=t[x].tms;
if(!t[t[x].fail].in)q.push(t[x].fail);
for(int i=0;i<t[x].fin.size();i++)occ[t[x].fin[i]]=t[x].tms;
}
}
int main(){
scanf("%d",&n);
for(register int i=1;i<=n;i++)scanf("%s",s+S[i-1]),S[i]=S[i-1]+strlen(s+S[i-1]),ins(i);
build();
for(register int i=1;i<=n;i++)merge(i);
topo();
for(register int i=1;i<=n;i++)printf("%d
",occ[i]);
return 0;
}