BZOJ 3473

思路：

CF原题

ZYF有题解

O(nlog^2n)

//By SiriusRen
#include <bits/stdc++.h>
using namespace std;
const int N=;
int n,m,q,cntA[N],cntB[N],A[N],B[N],rk[N],ht[N],sa[N],tsa[N],f[N][];
int from[N],cnt[N],rec[N],tl[N],ans[N];
char ch[N],s[N];
void SA(){
    for(int i=;i<=n;i++)cntA[s[i]]++;
    for(int i=;i<=;i++)cntA[i]+=cntA[i-];
    for(int i=n;i;i--)sa[cntA[s[i]]--]=i;
    rk[sa[]]=;
    for(int i=;i<=n;i++)rk[sa[i]]=rk[sa[i-]]+(s[sa[i]]!=s[sa[i-]]);
    for(int l=;rk[sa[n]]<n;l<<=){
        memset(cntA,,sizeof(cntA));
        memset(cntB,,sizeof(cntB));
        for(int i=;i<=n;i++)cntA[A[i]=rk[i]]++,cntB[B[i]=(i+l<=n?rk[i+]:)]++;
        for(int i=;i<=n;i++)cntA[i]+=cntA[i-],cntB[i]+=cntB[i-];
        for(int i=n;i;i--)tsa[cntB[B[i]]--]=i;
        for(int i=n;i;i--)sa[cntA[A[tsa[i]]]--]=tsa[i];
        rk[sa[]]=;
        for(int i=;i<=n;i++)rk[sa[i]]=rk[sa[i-]]+(A[sa[i]]!=A[sa[i-]]||B[sa[i]]!=B[sa[i-]]);
    }
    for(int i=,j=;i<=n;i++){
        j=j?j-:;
        while(s[i+j]==s[sa[rk[i]-]+j])j++;
        ht[rk[i]]=j;
    }
    for(int i=;i<=n;i++)f[i][]=ht[i];
    for(int j=;j<=;j++)
        for(int i=;i<=n;i++)
            f[i][j]=min(f[i][j-],f[i+(<<(j-))][j-]);
}
bool check(int pos,int len){
    int l=pos,r=pos;
    for(int j=;~j;j--){
        if(l+>=(<<j)&&f[l+-(<<j)][j]>=len)l-=(<<j);
        if(f[r+][j]>=len)r+=(<<j);
    }return rec[r]>=l;
}
int main(){
    scanf("%d%d",&m,&q);
    for(int i=;i<=m;i++){
        scanf("%s",ch);
        int t=strlen(ch);
        s[n++]=' ';
        for(int j=;j<t;j++)from[n]=i,s[n++]=ch[j];
        tl[i]=n;
    }n--,s[]=;SA();
    int t=,k=;
    for(int i=;i<=n;i++)if(from[sa[i]]){
        if(!cnt[from[sa[i]]])k++;
        cnt[from[sa[i]]]++;
        if(k>=q){
            for(;k-(cnt[from[sa[t]]]==)>=q;k-=(cnt[from[sa[t]]]==),--cnt[from[sa[t++]]]);
            rec[i]=t;
        }
    }
    for(int i=;i<=n;i++)if(from[sa[i]]){
        int l=,r=tl[from[sa[i]]]-sa[i],dt=;
        while(l<=r){
            int mid=(l+r)>>;
            if(check(i,mid))dt=mid,l=mid+;
            else r=mid-;
        }ans[from[sa[i]]]+=dt;
    }
    for(int i=;i<=m;i++)printf("%d ",ans[i]);
}