spoj 694 Distinct Substrings 后缀数组

根据height数组的性质，按height数组连续相邻的两个后缀之间是有重叠部分的，
所以扫一遍height数组，总子串的数量把重叠的部分减掉就行

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cstring>
#define maxn 1005
using namespace std;
int wa[maxn],wb[maxn],wv[maxn],wc[maxn];
int sa[maxn],ran[maxn],height[maxn];
int n;
char s[maxn];
int cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void mda(char *r,int *sa,int n,int m)
{
     int p,*x = wa,*y = wb,*t,j,i;
     for(i=0;i<m;i++)wc[i] = 0;
     for(i=0;i<n;i++)wc[x[i] = r[i]]++;
     for(i=1;i<m;i++)wc[i] += wc[i-1];
     for(i=n-1;i>=0;i--)sa[--wc[x[i]]] = i;

     for(p=1,j=1;p<n;j*=2,m =p)
     {
         for(p = 0,i=n-j;i<n;i++)y[p++] = i;
         for(i=0;i<n;i++) if(sa[i]>=j)y[p++] = sa[i]-j;

         for(i=0;i<n;i++)wv[i] = x[y[i]];

         for(i=0;i<m;i++)wc[i] = 0;
         for(i=0;i<n;i++)wc[wv[i]]++;
         for(i=1;i<m;i++)wc[i] += wc[i-1];
         for(i=n-1;i>=0;i--)sa[--wc[wv[i]]] = y[i];

         for(t=x,x=y,y=t,p=1,i=1,x[sa[0]] = 0;i<n;i++)
             x[sa[i]] = cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
}
void getHeight(char *r,int n)
{
    int k=0;
    for(int i=0;i<=n;i++)ran[sa[i]] = i;
    for(int i=0;i<n;i++)
    {
        if(k)k--;
        else k = 0;
        int j = sa[ran[i]-1];
        while(r[i+k]==r[j+k])k++;
        height[ran[i]] = k;
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%s",s);
       n = strlen(s);
       int ans = n*(n+1)/2;
       s[n] = 0;
       s[n+1] = 0;
       mda(s,sa,n+1,maxn);
       getHeight(s,n);
       for(int i=0;i<=n;i++)ans-=height[i];
       printf("%d\n",ans);
    }
    return 0;
}