这题就比较有趣了。
首先匹配一遍,然后统计子树叶子节点中包含大于等于k的节点个数(HH的项链)
然后就可以搞了。
关于合法的情况数,显然是l[i]-l[fa[i]],然后向下下传即可(YY一下)。
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i>=k;--i)
#define ll long long
#define maxn 600005
struct Query{
int l,r,id;
void print(){printf("Query : %d to %d (id is %d)\n",l,r,id);}
}brr[maxn];
bool cmp(Query a,Query b)
{return a.l==b.l?a.r<b.r:a.l<b.l;}
struct Bit_Tree{
int x[maxn];
void add(int pos,int f,int tot)
{for (;pos<=tot;pos+=(pos&(-pos))) x[pos]+=f;}
int query(int pos)
{int ret=0;for(;pos;pos-=(pos&(-pos)))ret+=x[pos];return ret;}
}BT;
class Suffix_Automata{
public:
int arr[maxn],tot,top;
int last,cnt,n,k;
char s[maxn];
vector <int> v[maxn],vb[maxn];
int h[maxn],to[maxn],ne[maxn],en,in[maxn],out[maxn];
int go[maxn][26],l[maxn],fa[maxn];
void init()
{
last=cnt=1; en=0;
memset(h,-1,sizeof h);
memset(go,0,sizeof go);
}
void add(int x,int id)
{
int p=last,q;
if (q=go[p][x])
{
if (l[q]==l[p]+1) last=q;
else
{
int nq=++cnt;
l[nq]=l[p]+1;
memcpy(go[nq],go[q],sizeof go[q]);
fa[nq]=fa[q];
fa[q]=nq;
for(;p&&go[p][x]==q;p=fa[p]) go[p][x]=nq;
last=nq;
}
}
else
{
int np=++cnt; l[np]=l[p]+1;
for (;p&&!go[p][x];p=fa[p]) go[p][x]=np;
if (!p) fa[np]=1;
else
{
q=go[p][x];
if (l[q]==l[p]+1) fa[np]=q;
else
{
int nq=++cnt;
l[nq]=l[p]+1;
memcpy(go[nq],go[q],sizeof go[q]);
fa[nq]=fa[q];
fa[q]=fa[np]=nq;
for (;p&&go[p][x]==q;p=fa[p]) go[p][x]=nq;
}
}
last=np;
}
v[last].push_back(id);
vb[id].push_back(last);
}
void addedge(int a,int b)
{
to[en]=b;
ne[en]=h[a];
h[a]=en++;
}
void ins(int id)
{
last=1;scanf("%s",s+1);
int len=strlen(s+1);
F(j,1,len) add(s[j]-'a',id);
}
void dfs(int o)
{
in[o]=++tot;
for (int i=0;i<v[o].size();++i) arr[tot++]=v[o][i];
for (int i=h[o];i>=0;i=ne[i]) dfs(to[i]);
out[o]=tot;
++top;brr[top].l=in[o];brr[top].r=out[o];brr[top].id=o;
}
void dfs2(int o)
{
for (int i=h[o];i>=0;i=ne[i])
f[to[i]]+=f[o],dfs2(to[i]);
}
int lst[maxn],nxt[maxn],ans[maxn],f[maxn];
void solve()
{
init();
scanf("%d%d",&n,&k);
F(i,1,n) ins(i);
F(i,1,cnt) addedge(fa[i],i);
dfs(1);
// printf("Arr : ");F(i,1,tot) printf("%d ",arr[i]); printf("\n");
sort(brr+1,brr+top+1,cmp);
// F(i,1,top) brr[i].print();
F(i,1,tot)
{
if (!lst[arr[i]]&&arr[i]) BT.add(i,1,tot);
else nxt[lst[arr[i]]]=i;
lst[arr[i]]=i;
}
// F(i,1,tot) printf("%d ",nxt[i]); printf("\n");
int now=1;
F(i,1,top)
{
while (now<brr[i].l)
{
if (nxt[now]&&arr[nxt[now]]) BT.add(nxt[now],1,tot);
now++;
}
ans[brr[i].id]=BT.query(brr[i].r)-BT.query(brr[i].l-1);
}
// F(i,1,top) printf("[%d] %d\n",i,ans[i]);
F(i,1,top) if (ans[i]>=k) f[i]=l[i]-l[fa[i]]; else f[i]=0;
dfs2(1);
F(i,1,n)
{
ll ans=0;
F(j,0,vb[i].size()-1) ans+=f[vb[i][j]];
printf("%lld",ans);
if (i!=n) printf(" ");
}
}
}sam;
int main()
{sam.solve();}