每个子串一定是某个后缀的前缀,那么原问题等价于求所有后缀之间的不相同的前缀的个数。如果所有的后缀按照 suffix(sa[1]), suffix(sa[2]),suffix(sa[3]), …… ,suffix(sa[n])的顺序计算,不难发现,对于每一次新加进来的后缀 suffix(sa[k]),它将产生 n-sa[k]+1 个新的前缀。但是其中有height[k]个是和前面的字符串的前缀是相同的。所以 suffix(sa[k])将“贡献”出 n-sa[k]+1- height[k]个不同的子串。累加后便是原问题的答案。这个做法的时间复杂度为 O(n)
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=1010;
int a[maxn],b[maxn],s[maxn],c[maxn];
int *rank,*height,*sa=s+1;
void sortandrank(int *a1,int *a2,int n,int &m,int j)
{
int i;memset(c,0,sizeof(c));
for(i=0;i<n;i++) c[a1[i]]++;
for(i=1;i<=m;i++) c[i]+=c[i-1];
for(i=n-1;i>=0;i--) sa[--c[a1[a2[i]]]]=a2[i];
a2[sa[0]]=m=0;
for(i=1;i<n;i++) a2[sa[i]]=a1[sa[i-1]]==a1[sa[i]]&&a1[sa[i-1]+j]==a1[sa[i]+j]?m:++m;
}
void da(char*str,int n,int m)
{
int *a1=a,*a2=b,*tmp;
int i,j,p;
for(i=0;i<n;i++) a1[i]=i,a2[i]=str[i];
a1[n]=a2[n]=-1;
sortandrank(a2,a1,n,m,0);
for(j=1;m<n-1;j<<=1)
{
p=0;
for(i=n-j;i<n;i++) a2[p++]=i;
for(i=0;i<n;i++) if(sa[i]>=j) a2[p++]=sa[i]-j;
sortandrank(a1,a2,n,m,j);
tmp=a1,a1=a2,a2=tmp;
}
rank=a1,height=a2;
}
void calheight(char*str,int n)
{
int i,j,k;
sa[-1]=n;
for(height[0]=k=i=0;i<n;i++)
{
for(k?k--:0,j=sa[rank[i]-1];str[i+k]==str[j+k];k++);
height[rank[i]]=k;
}
}
char str[maxn];
int main()
{
int n;cin.sync_with_stdio(false);
cin>>n;
while(n--)
{
cin>>str;
int len=strlen(str);str[len]=0;
da(str,len+1,128);calheight(str,len);
int ans=0;
for(int i=1;i<=len;i++)
{
ans+=len-sa[i]-height[i];
}
cout<<ans<<endl;
}
return 0;
}