spoj 694 Distinct Substrings 后缀数组

来源：互联网发布：linux 安装squid 编辑：程序博客网时间：2024/06/07 01:43

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

#include<cstdio>#include<cstring>#include<algorithm>#include<iostream>#include<cstring>#define maxn 1005using 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;}

0 0