SPOJ 694 DISUBSTR - Distinct Substrings

思路

求本质不同的子串个数，总共重叠的子串个数就是height数组的和

总子串个数-height数组的和即可

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
#define int long long
const int MAXN = 100000;
using namespace std;
int height[MAXN],sa[MAXN],ranks[MAXN],barrel[MAXN],n;
struct Node{
    int pos,r[2];
}midx[MAXN],x[MAXN];
char s[MAXN];
int c_sort(int n,int lim){
    for(int i=0;i<2;i++){
        memset(barrel,0,sizeof(barrel));
        for(int j=1;j<=n;j++)
            barrel[x[j].r[i]]++;
        for(int j=1;j<=lim;j++)
            barrel[j]+=barrel[j-1];
        for(int j=n;j>=1;j--)
            midx[barrel[x[j].r[i]]--]=x[j];
        for(int j=1;j<=n;j++)
            x[j]=midx[j];
    }
    ranks[x[1].pos]=1;
    int cnt=1;
    for(int i=2;i<=n;i++){
        if(x[i].r[0]==x[i-1].r[0]&&x[i].r[1]==x[i-1].r[1])
            ranks[x[i].pos]=cnt;
        else
            ranks[x[i].pos]=++cnt;
    }
    return cnt;
}
void cal_sa(int n){
    for(int i=1;i<=n;i++)
        x[i]=(Node){i,s[i],0};
    int cnt=c_sort(n,1500);
    for(int i=1;cnt<n;i<<=1){
        for(int j=1;j<=n;j++)
            x[j]=(Node){j,(i+j<=n)?ranks[i+j]:0,ranks[j]};
        cnt=c_sort(n,cnt);
    }
    for(int i=1;i<=n;i++)
        sa[ranks[i]]=i;
    for(int i=1,j=0,k;i<=n;height[ranks[i++]]=j)
        for(j?j--:0,k=sa[ranks[i]-1];s[i+j]==s[j+k];j++);
}
void init(void){
   memset(sa,0,sizeof(sa));
   memset(height,0,sizeof(height));
   memset(ranks,0,sizeof(ranks));
   memset(x,0,sizeof(x));
   memset(midx,0,sizeof(midx));
}
signed main(){
    int T;
    scanf("%lld",&T);
    while(T--){
        init();
        scanf("%s",s+1);
        n=strlen(s+1);
        cal_sa(n);
        int sum=0;
        for(int i=2;i<=n;i++)
            sum+=height[i];
        printf("%lld\n",(n+1)*n/2-sum);
    }
    return 0;
}