CSU-1632 Repeated Substrings[后缀数组求重复出现的子串数目]

评测地址：https://cn.vjudge.net/problem/CSU-1632

Description

求字符串中所有出现至少2次的子串个数

Input

第一行为一整数T(T<=10)表示用例组数，每组用例占一行为一个长度不超过100000的字符串

Output

对于每组用例，输出该串中所有出现至少两次的子串个数

Sample Input

3

aabaab

aaaaa

AaAaA

Sample Output

5

4

5

Solution

Ans=sum（max(height(i)-height(i-1),0)）

#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e5+;
int T,n,ans,c[N],sa[N],tsa[N],trank[N],rank[N],h[N];
char s[N];
void DA(int maxx=){
    memset(c,,sizeof c);int p;
    for(int i=;i<=n;i++) c[rank[i]=s[i]]++;
    for(int i=;i<=maxx;i++) c[i]+=c[i-];
    for(int i=n;i;i--) sa[c[rank[i]]--]=i;
    trank[sa[]]=p=;
    for(int i=;i<=n;i++){
        if(rank[sa[i]]!=rank[sa[i-]]) p++;
        trank[sa[i]]=p;
    }
    for(int i=;i<=n;i++) rank[i]=trank[i];
    for(int k=;p<n;k<<=,maxx=p){
        p=;
        for(int i=n-k+;i<=n;i++) tsa[++p]=i;
        for(int i=;i<=n;i++) if(sa[i]>k) tsa[++p]=sa[i]-k;
        memset(c,,sizeof c);
        for(int i=;i<=n;i++) trank[i]=rank[tsa[i]];
        for(int i=;i<=n;i++) c[trank[i]]++;
        for(int i=;i<=maxx;i++) c[i]+=c[i-];
        for(int i=n;i;i--) sa[c[trank[i]]--]=tsa[i];
        trank[sa[]]=p=;
        for(int i=;i<=n;i++){
            if(rank[sa[i]]!=rank[sa[i-]]||rank[sa[i]+k]!=rank[sa[i-]+k]) p++;
            trank[sa[i]]=p;
        }
        for(int i=;i<=n;i++) rank[i]=trank[i];
    }
    for(int i=,k=;i<=n;i++){
        int j=sa[rank[i]-];
        while(s[i+k]==s[j+k]) k++;
        h[rank[i]]=k;if(k>)k--;
    }
}
void GO(){
    ans=;
    for(int i=;i<=n;i++) if(h[i]>h[i-]) ans+=h[i]-h[i-];
    printf("%d\n",ans);
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%s",s+);n=strlen(s+);
        DA();
        GO();
    }
    return ;
}