后缀自动机版本:

所有的串用(char)('z'+1)连起来,然后建自动机。再用原串在自动机上跑。跑到的位置的endpos就是出现的次数。不过内存有点大。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cstring>
#include <map>
#include <queue>
#include <set>
#include <cassert>
#include <stack>
#include <bitset>
#define mkp make_pair
using namespace std;
const double EPS=1e-;
typedef long long lon;
const lon SZ=,SSZ=*SZ,APB=,one=,INF=0x7FFFFFFF,mod=;
int n,len,cnt,slink[SSZ],nex[SSZ][APB],maxlen[SSZ];
int minlen[SSZ],endpos[SSZ],in[SSZ];
char ch[SZ];
string str[SZ];
bool green[SSZ]; void add(string &x)
{
for(int i=;i<x.size();++i)
{
ch[++len]=x[i];
}
ch[++len]='z'+;
} int ins(int pre,int c)
{
int z=++cnt,u=pre;
green[z]=;
for(;u!=-&&!nex[u][c];u=slink[u])nex[u][c]=z;
maxlen[z]=maxlen[pre]+;
if(u==-)
{
slink[z]=;
}
else
{
int x=nex[u][c];
if(maxlen[x]==maxlen[u]+)
{
slink[z]=x;
}
else
{
int v=++cnt;
memcpy(nex[v],nex[x],sizeof(nex[x]));
maxlen[v]=maxlen[u]+;
slink[v]=slink[x];
minlen[v]=maxlen[slink[v]]+;
slink[x]=slink[z]=v;
minlen[x]=maxlen[v]+;
for(;u!=-&&nex[u][c]==x;u=slink[u])
nex[u][c]=v;
}
}
minlen[z]=maxlen[slink[z]]+;
return z;
} void topo()
{
for(int i=;i<=cnt;++i)
{
++in[slink[i]];
}
stack<int> stk;
for(int i=;i<=cnt;++i)
{
if(!in[i])stk.push(i),endpos[i]=;
}
for(;stk.size();)
{
int top=stk.top();
stk.pop();
int pre=slink[top];
endpos[pre]+=endpos[top],--in[pre];
if(!in[pre]&&pre)
{
if(green[pre])++endpos[pre];
stk.push(pre);
}
}
} void init()
{
cin>>n;
for(int i=;i<=n;++i)
{
cin>>str[i];
add(str[i]);
}
int pre=;
slink[]=-;
for(int i=;i<=len;++i)
{
pre=ins(pre,ch[i]-'a');
}
topo();
} int calc(string &x)
{
int cur=;
for(int i=;i<x.size();++i)
{
int c=x[i]-'a';
cur=nex[cur][c];
}
return endpos[cur];
} void work()
{
for(int i=;i<=n;++i)
{
cout<<calc(str[i])<<endl;
}
} int main()
{
std::ios::sync_with_stdio();
//freopen("d:\\1.txt","r",stdin);
int casenum;
//cin>>casenum;
//cout<<casenum<<endl;
//for(int time=1;time<=casenum;++time)
//for(int time=1;cin>>n>>qnum,n;++time)
{
//cout<<"Case "<<time<<": ";
init();
work();
}
return ;
}
05-07 15:33