洛谷P3966 [TJOI2013]单词(后缀自动机)

时间:2023-03-09 09:15:43
洛谷P3966 [TJOI2013]单词(后缀自动机)

传送门

统计单词出现次数……为啥大家都是写AC自动机的嘞……明明后缀自动机也能做的说……

统计出现次数这个就直接按长度排序然后做个dp就好,这是SAM的板子的要求啊,不提了

然后考虑怎么让所有串之间隔开。本来打算建个广义SAM后来发现没办法处理子串重复的情况……然后就按题解里的方法在每两个串之间加入一个新字符然后直接对整个串建好了

 //minamoto
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=4e6+;
int fa[N],ch[N][],l[N],cnt[N],last=,tot=;
char s[N>>],t[N>>];int len[N>>],c[N],a[N],k,ans[N],n;
void ins(int c){
int p=last,np=++tot;last=np,l[np]=l[p]+,cnt[np]=;
for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
if(!p) fa[np]=;
else{
int q=ch[p][c];
if(l[q]==l[p]+) fa[np]=q;
else{
int nq=++tot;l[nq]=l[p]+;
memcpy(ch[nq],ch[q],sizeof(ch[q]));
fa[nq]=fa[q],fa[q]=fa[np]=nq;
for(;ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
}
}
}
inline void calc(){
for(int i=;i<=tot;++i) ++c[l[i]];
for(int i=;i<=tot;++i) c[i]+=c[i-];
for(int i=;i<=tot;++i) a[c[l[i]]--]=i;
for(int i=tot;i;--i) cnt[fa[a[i]]]+=cnt[a[i]];
}
int main(){
// freopen("testdata.in","r",stdin);
scanf("%d",&n);
for(int i=;i<=n;++i){
scanf("%s",t+);len[i]=strlen(t+);
for(int j=;j<=len[i];++j) ins(s[++k]=t[j]-'a');
ins(s[++k]=);
}
calc();k=;
for(int i=,x,j;i<=n;++i){
for(j=,x=;j<=len[i];++j)
x=ch[x][s[++k]];
ans[i]=cnt[x];
++k;
}
for(int i=;i<=n;++i) printf("%d\n",ans[i]);
return ;
}