根据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;
}