Distinct Substrings（spoj 694）

题意：要求不同子串的个数

/*
    先求出height数组，不难看出height之和就是重复的字符串个数，用总的减去它就行了。
*/
#include<cstdio>
#include<iostream>
#include<cstring>
#define N 100010
int sa[N],rk[N],height[N],t1[N],t2[N],c[N];
char s[N];
using namespace std;
bool cmp(int *y,int a,int b,int k){
    return y[a]==y[b]&&y[a+k]==y[b+k];
}
void DA(int n,int m){
    int *x=t1,*y=t2;
    for(int i=;i<m;i++) c[i]=;
    for(int i=;i<n;i++) c[x[i]=s[i]]++;
    for(int i=;i<m;i++) c[i]+=c[i-];
    for(int i=n-;~i;i--) sa[--c[x[i]]]=i;
    for(int k=,p=;k<=n;k*=,m=p,p=){
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=;i<m;i++) c[i]=;
        for(int i=;i<n;i++) c[x[y[i]]]++;
        for(int i=;i<m;i++) c[i]+=c[i-];
        for(int i=n-;~i;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);p=;x[sa[]]=;
        for(int i=;i<n;i++)
            if(cmp(y,sa[i-],sa[i],k)) x[sa[i]]=p-;
            else x[sa[i]]=p++;
        if(p>=n) break;
    }
    for(int i=;i<n;i++) rk[sa[i]]=i;
}
void get_ht(int n){
    for(int i=,j,k=;i<n;height[rk[i++]]=k){
        j=sa[rk[i]-];k=k?k-:;
        while(s[i+k]==s[j+k]) k++;
    }
}
int solve(int n){
    long long sum=1LL*n*(n+)/;
    for(int i=;i<=n;i++) sum-=(long long)height[i];
    return (int)sum;
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        scanf("%s",s);int n=strlen(s);
        DA(n+,);
        get_ht(n);
        printf("%d\n",solve(n));
    }
    return ;
}