传送门biu~biu~
对于所有串建广义后缀自动机,沿着父亲指针不断向root走,把路径上的节点计数+1。为了避免出现重复,记录每个节点最后一次被哪个串访问,串在走的时候如果遇到了这个串已经访问过的节点就停止。
可以在parent树上dp出这个点到root路径上有多少子串出现了k次以上作为答案。最后把每个串在自动机上跑一遍,过程中每个访问到的节点答案的和即为每个串的答案。
复杂度不会证。
#include<bits/stdc++.h>
using namespace std;
string s[100005];
int n,k;
struct Node{
vector<Node*>to;
Node *ch[26],*fa;
int val,sum,vis;
long long ans;
Node(int _=0):val(_){
sum=vis=ans=0,fa=0x0;
memset(ch,0x0,sizeof ch);
}
}*root=new Node,*last;
inline void extend(int z){
Node *p=last,*np=last=new Node(p->val+1);
while(p && !p->ch[z]) p->ch[z]=np,p=p->fa;
if(!p){
np->fa=root;
root->to.push_back(np);
return;
}
Node *q=p->ch[z];
if(q->val==p->val+1){
np->fa=q;
q->to.push_back(np);
return;
}
Node *nq=new Node(p->val+1);
memcpy(nq->ch,q->ch,sizeof nq->ch);
nq->fa=q->fa;q->fa->to.push_back(nq);
q->fa=np->fa=nq;nq->to.push_back(q),nq->to.push_back(np);
while(p && p->ch[z]==q) p->ch[z]=nq,p=p->fa;
}
void dfs(Node *o){
if(o!=root){
o->ans=(o->sum>=k)*(o->val-o->fa->val);
o->ans+=o->fa->ans;
}
for(int i=0;i<o->to.size();++i)
if(o->to[i]->fa==o) dfs(o->to[i]);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;++i){
cin>>s[i],last=root;
for(int j=0;s[i][j];++j) extend(s[i][j]-'a');
}
for(int i=1;i<=n;++i){
Node *o=root;
for(int j=0;s[i][j];++j){
o=o->ch[s[i][j]-'a'];
Node *p=o;
while(p && p->vis!=i) p->vis=i,++p->sum,p=p->fa;
}
}
dfs(root);
for(int i=1;i<=n;++i){
Node *o=root;long long ans(0);
for(int j=0;s[i][j];++j)
o=o->ch[s[i][j]-'a'],ans+=o->ans;
printf("%lld ",ans);
}
return 0;
}