spoj 694&&705 后缀数组

每个子串一定是某个后缀的前缀，那么原问题等价于求所有后缀之间的不相同的前缀的个数。如果所有的后缀按照 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;
}