BZOJ3277 串（后缀数组+二分答案+主席树）

　　因为不会SAM，考虑SA。将所有串连起来并加分隔符，每次考虑计算以某个位置开始的子串有多少个合法。

　　对此首先二分答案，找到名次数组上的一个区间，那么只需要统计有多少个所给串在该区间内出现就可以了。这是一个主席树的经典问题，对每个数找到上次出现位置扔进去即可。这样就做到O(nlog²n)了。

　　可以进一步优化到O(nlogn)，不是很会。

　　以及我又不会写SA了，没救。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
    int x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
#define N 200010
int n,m,tot,pos[N],cnt[N],end[N],sa[N],sa2[N],rk[N<<],tmp[N<<],h[N],lg2[N],f[N][],root[N],pre[N],p[N],cnt2=;
long long ans[N];
struct data{int l,r,x;
}tree[N<<];
char s[N],c[N];
void make(int n)
{
    int m=;
    for (int i=;i<=n;i++) cnt[rk[i]=s[i]]++,m=max(m,(int)s[i]);
    for (int i=;i<=m;i++) cnt[i]+=cnt[i-];
    for (int i=n;i>=;i--) sa[cnt[rk[i]]--]=i;
    for (int k=;k<=n;k<<=)
    {
        int p=;
        for (int i=n-k+;i<=n;i++) sa2[++p]=i;
        for (int i=;i<=n;i++) if (sa[i]>k) sa2[++p]=sa[i]-k;
        memset(cnt,,sizeof(cnt));
        for (int i=;i<=n;i++) cnt[rk[i]]++;
        for (int i=;i<=m;i++) cnt[i]+=cnt[i-];
        for (int i=n;i>=;i--) sa[cnt[rk[sa2[i]]]--]=sa2[i];
        memcpy(tmp,rk,sizeof(rk));
        p=rk[sa[]]=;
        for (int i=;i<=n;i++)
        {
            if (tmp[sa[i]]!=tmp[sa[i-]]||tmp[sa[i]+k]!=tmp[sa[i-]+k]) p++;
            rk[sa[i]]=p;
        }
        m=p;if (m>=n) break;
    }
    for (int i=;i<=n;i++)
    {
        h[i]=max(h[i-]-,);
        while (s[i+h[i]]==s[sa[rk[i]-]+h[i]]) h[i]++;
    }
    for (int i=;i<=n;i++) f[i][]=h[sa[i]];
    lg2[]=;
    for (int i=;i<=n;i++)
    {
        lg2[i]=lg2[i-];
        if ((<<lg2[i])<=i) lg2[i]++;
    }
    for (int j=;j<=;j++)
        for (int i=;i<=n;i++)
        f[i][j]=min(f[i][j-],f[min(i+(<<j-),n)][j-]);
}
int query(int x,int y)
{
    if (x>y) swap(x,y);
    x++;if (x>y) return N;
    return min(f[x][lg2[y-x+]],f[y-(<<lg2[y-x+])+][lg2[y-x+]]);
}
void ins(int &k,int l,int r,int x)
{
    tree[++cnt2]=tree[k],k=cnt2;tree[k].x++;
    if (l==r) return;
    int mid=l+r>>;
    if (x<=mid) ins(tree[k].l,l,mid,x);
    else ins(tree[k].r,mid+,r,x);
}
bool Query(int x,int y,int l,int r,int p,int m)
{
    if (m<=) return ;
    if (!y) return ;
    if (l==r) return m<=tree[y].x-tree[x].x;
    int mid=l+r>>;
    if (p<=mid) return Query(tree[x].l,tree[y].l,l,mid,p,m);
    else return Query(tree[x].r,tree[y].r,mid+,r,p,m-tree[tree[y].l].x+tree[tree[x].l].x);
}
void build()
{
    for (int i=;i<=tot;i++) pre[i]=p[pos[sa[i]]],p[pos[sa[i]]]=i;
    for (int i=;i<=tot;i++)
    {
        root[i]=root[i-];
        ins(root[i],,tot,pre[i]);
    }
}
bool check(int x,int k,int m)
{
    int p,q,l=,r=x;
    while (l<=r)
    {
        int mid=l+r>>;
        if (query(mid,x)>=k) p=mid,r=mid-;
        else l=mid+;
    }
    l=x,r=tot;
    while (l<=r)
    {
        int mid=l+r>>;
        if (query(mid,x)>=k) q=mid,l=mid+;
        else r=mid-;
    }
    return Query(root[p-],root[q],,tot,p-,m);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj3277.in","r",stdin);
    freopen("bzoj3277.out","w",stdout);
    const char LL[]="%I64d ";
#else
    const char LL[]="%lld ";
#endif
    n=read(),m=read();
    for (int i=;i<=n;i++)
    {
        scanf("%s",c+);int m=strlen(c+);
        for (int j=;j<=m;j++) s[++tot]=c[j],pos[tot]=i;
        s[++tot]='$',pos[tot]=i;end[i]=tot;
    }
    make(tot);
    build();
    for (int i=;i<=tot;i++)
    if (s[i]!='$')
    {
        int l=,r=end[pos[i]]-i,x=;
        while (l<=r)
        {
            int mid=l+r>>;
            if (check(rk[i],mid,m)) x=mid,l=mid+;
            else r=mid-;
        }
        ans[pos[i]]+=x;
    }
    for (int i=;i<=n;i++) printf(LL,ans[i]);
    return ;
}