bzoj3277: 串

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<string>
#define maxn 400005
#define ll long long
using namespace std;
int n,m,rt=1,la=1,cnt=1;
int tax[maxn],ord[maxn];
ll f[maxn];
char ch[maxn];string a[maxn];
struct node{
    int Max,par,nex[26];
    int now,num;
}s[maxn*2];

void ins(int c){
    int np=++cnt,p=la;la=np;
    s[np].Max=s[p].Max+1;
    for(;p&&!s[p].nex[c];p=s[p].par)s[p].nex[c]=np;
    if(!p)s[np].par=rt;
    else {
        int q=s[p].nex[c];
        if(s[q].Max==s[p].Max+1)s[np].par=q;
        else {
            int nq=++cnt;
            s[nq].Max=s[p].Max+1;
            for(int i=0;i<26;i++)s[nq].nex[i]=s[q].nex[i];
            s[nq].par=s[q].par;
            s[q].par=s[np].par=nq;
            for(;s[p].nex[c]==q;p=s[p].par)s[p].nex[c]=nq;    
        }
    }
}

void Sort(){
    for(int i=1;i<=cnt;i++)tax[s[i].Max]++;
    for(int i=1;i<=cnt;i++)tax[i]+=tax[i-1];
    for(int i=1;i<=cnt;i++)ord[tax[s[i].Max]--]=i;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        scanf("%s",ch);
        a[i]=(string)ch;la=rt;
        for(int j=0;j<a[i].size();j++){
            ins(a[i][j]-‘a‘);
        }
    }
    for(int i=1;i<=n;i++){
        int k=rt;
        for(int j=0;j<a[i].size();j++){
            k=s[k].nex[a[i][j]-‘a‘];
            int p=k;
            for(;s[p].now!=i&&p;p=s[p].par)s[p].num++,s[p].now=i;
        }
    }
    Sort();
    s[0].Max=0;
    for(int i=1;i<=cnt;i++){
        int k=ord[i];
        f[k]=f[s[k].par]+(s[k].num>=m?-s[s[k].par].Max+s[k].Max:0);
    }
    for(int i=1;i<=n;i++){
        ll ans=0;int k=rt;
        for(int j=0;j<a[i].size();j++){
            k=s[k].nex[a[i][j]-‘a‘];
            ans+=f[k];
        }
        printf("%lld\n",ans);
    }
    return 0;
}