menu zcmimi's blog
LG 2408 不同子串个数
查看原题

点击跳转

\frac {n(n+1)}2 - \sum height[i]

#include<bits/stdc++.h>
const int N=100011;
char s[N];
int n,m,sa[N],rnk[N],tax[N],tp[N];
void qsort(){
    for(int i=0;i<=m;++i)tax[i]=0;
    for(int i=1;i<=n;++i)++tax[rnk[i]];
    for(int i=1;i<=m;++i)tax[i]+=tax[i-1];
    for(int i=n;i;--i)sa[tax[rnk[tp[i]]]--]=tp[i];
}
void suffix(){
    m=100;
    for(int i=1;i<=n;++i)sa[i]=0,rnk[i]=s[i]-'a'+1,tp[i]=i;
    qsort();
    for(int w=1,p=0;p<n;m=p,w<<=1){
        p=0;
        for(int i=1;i<=w;++i)tp[++p]=n-w+i;
        for(int i=1;i<=n;++i)
            if(sa[i]>w)tp[++p]=sa[i]-w;
        qsort();
        std::swap(tp,rnk);
        rnk[sa[1]]=p=1;
        for(int i=2;i<=n;++i)
            rnk[sa[i]]=p+=tp[sa[i]]^tp[sa[i-1]]||tp[sa[i]+w]^tp[sa[i-1]+w];
    }
}
int main(){
    scanf("%d%s",&n,s+1);
    suffix();
    long long ans=1ll*n*(n+1)>>1;
    for(int i=1,k=0;i<=n;++i){
        if(k)--k;
        int j=sa[rnk[i]-1];
        while(s[i+k]==s[j+k]&&i+k<=n&&j+k<=n)++k;
        ans-=k;
    }
    printf("%lld\n",ans);
}
LG 2408 不同子串个数

评论加载中...

Search
search