bzoj3277: 串

Description

字符串是oi界常考的問題。現在給定你n個字符串，詢問每個字符串有多少子串（不包括空串）是所有n個字符串中

至少k個字符串的子串（注意包括本身）。

Input

第一行兩個整數n，k。

接下來n行每行一個字符串。

n,k,l<=100000

Output

輸出一行n個整數，第i個整數表示第i個字符串的答案。

Sample Input

3 1
abc
a
ab

Sample Output

6 1 3

---

Sol

多個串可以建廣義sam解決。

每次加入新串時就把la=rt就行。

全部建完後，從每個串的每個位置往上跳，統計每個點出現幾次

dp F[i] 表示i和i的後綴出現次數>=k的有幾個。

f[i]=f[par]+(num[i]>=k)?Max[par]-Max[i]:0;

把串的每個前綴當成累加就行。

#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;
}