Repeated Substrings（UVAlive 6869）

题意：求出现过两次以上的不同子串有多少种。

/*
    用后缀数组求出height[]数组，然后扫一遍，
    发现height[i]-height[i-1]>=0,就ans+=height[i]-height[i-1]。
*/
#include<cstdio>
#include<iostream>
#include<cstring>
#define N 100010
using namespace std;
int sa[N],rk[N],ht[N],t1[N],t2[N],c[N],n,m=;
char s[N];
bool cmp(int *y,int a,int b,int k){
    int a1=y[a],b1=y[b];
    int a2=a+k>=n?-:y[a+k];
    int b2=b+k>=n?-:y[b+k];
    return a1==b1&&a2==b2;
}
void DA(){
    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;
    }
}
void get_ht(){
    for(int i=;i<n;i++) rk[sa[i]]=i;
    for(int i=,k=;i<n;ht[rk[i++]]=k){
        if(!rk[i]) continue;
        int j=sa[rk[i]-];k=k?k-:k;
        while(j+k<n&&i+k<n&&s[j+k]==s[i+k]) k++;
    }
    ht[]=;
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        memset(s,,sizeof(s));
        memset(ht,,sizeof(ht));
        m=;
        scanf("%s",s);
        n=strlen(s);
        DA();get_ht();
        long long ans=;
        for(int i=;i<n;i++)
            if(ht[i]>ht[i-]) ans+=(long long)(ht[i]-ht[i-]);
        cout<<ans<<endl;
    }
    return ;
}