bzoj5137 [Usaco2017 Dec]Standing Out from the Herd（后缀数组）

最新推荐文章于 2021-03-23 14:19:33 发布

原创最新推荐文章于 2021-03-23 14:19:33 发布 · 599 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

bzoj 同时被 3 个专栏收录

764 篇文章

订阅专栏

--------字符串--------

114 篇文章

订阅专栏

37 篇文章

订阅专栏

本文介绍了一种使用SA（Suffix Array，后缀数组）算法来计算一组字符串中本质不同的子串数量的方法。通过将多个字符串串联并利用后缀数组进行处理，可以有效地找出这些字符串中的所有独特子串，并考虑到各个字符串间的重复子串问题。

我们首先把所有串接在一起，用分隔符隔开，跑一遍sa，求出h数组。然后我们考虑每个串如何算贡献，一个串的所有后缀一定是在rank数组上的若干连续段，对于每一个连续段[l,r]，我们首先计算出本质不同的子串个数，类似这题。然后考虑因出现在其他串而要减少的贡献，就是lcp(l-1,l)+lcp(r,r+1)-lcp(l-1,r+1)，注意去重。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int nn,n=0,m=100,sa[N],rk[N<<1],rk1[N],h[N],bel[N],R[N>>1],a[N];
int cnt[N],tmp[N];ll ans[N];
char s[N];
int main(){
//  freopen("a.in","r",stdin);
    nn=read();
    for(int i=1;i<=nn;++i){
        scanf("%s",s+1);int len=strlen(s+1);
        for(int j=1;j<=len;++j) a[++n]=s[j]-'a'+1,bel[n]=i;
        R[i]=n;a[++n]=++m;
    }for(int i=1;i<=n;++i) cnt[a[i]]=1;
    for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
    for(int i=1;i<=n;++i) rk[i]=cnt[a[i]];
    int k=0;
    for(int p=1;k!=n;p+=p,m=k){
        for(int i=1;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[i+p]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) tmp[cnt[rk[i+p]]--]=i;
        for(int i=1;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[tmp[i]]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) sa[cnt[rk[tmp[i]]]--]=tmp[i];
        memcpy(rk1,rk,sizeof(rk1));rk[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rk1[sa[i]]!=rk1[sa[i-1]]||rk1[sa[i]+p]!=rk1[sa[i-1]+p]) ++k;
            rk[sa[i]]=k;
        }
    }k=0;
    for(int i=1;i<=n;++i){
        if(rk[i]==1){h[1]=0;continue;}
        if(i==1||h[rk[i-1]]<=1) k=0;
        if(k) --k;
        while(a[i+k]==a[sa[rk[i]-1]+k]) ++k;
        h[rk[i]]=k;
    }int mn=0;
    for(int i=1;i<=n;++i){
        if(a[sa[i]]>=100) break;int x=bel[sa[i-1]],y=bel[sa[i]];
        mn=min(mn,h[i]);
        if(x!=y){
            ans[x]-=h[i];ans[x]+=mn;
            ans[y]-=h[i];mn=h[i];ans[y]+=R[y]-sa[i]+1;
        }else ans[x]+=R[x]-sa[i]+1-h[i];
    }for(int i=1;i<=nn;++i) printf("%lld\n",ans[i]);
    return 0;
}