SPOJ 694 不重复子串

按照sa数组的后缀字典序进行判断 每个串插入的时候应该有L个不同的子串（L为长度）

那么我们只要减去前面和它相同的重复前缀就可以算出加入这一条增加子串个数

那么就是 height[i]

论文水题 估计是要找变形了

#include <cstdio>
#include <cstring>
#define maxn 222222
#define ll long long
#define mid ((l+r)>>1)
char s[maxn];
int n,m,sa[maxn],rank[maxn],height[maxn];
int wa[maxn],wb[maxn],ws[maxn],wv[maxn];
int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
inline bool cmp(int *r,int a,int b,int l)
{
    return (r[a] == r[b]) && (r[a+l] == r[b+l]);
}
inline void da(int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    memset(ws,0,sizeof(ws));
    for (i=0;i<n;++i)  ws[x[i]=s[i]]++;
    for (i=1;i<m;++i)  ws[i]+=ws[i-1];
    for (i=n-1;i>=0;--i)  sa[--ws[x[i]]]=i;
    for (j=1,p=1;p<n;j<<=1,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)  ws[i]=0;
        for (i=0;i<n;++i)  ws[wv[i]]++;
        for (i=1;i<m;++i)  ws[i]+=ws[i-1];
        for (i=n-1;i>=0;--i)  sa[--ws[wv[i]]]=y[i];
        for (t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;++i)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
inline void calcheight()
{
    int i,j,k=0;
    for (i=1;i<=n;++i)  rank[sa[i]]=i;
    for (i=0;i<n;height[rank[i++]]=k)
        for (k?k--:0,j=sa[rank[i]-1];s[i+k]==s[j+k];k++);
}//sa:1--strlen(s+'#')=n
//height:2--n
void print(){
    for(int i=1;i<=n;++i){
        printf("%d %d ",i,sa[i]);
        for(int j=sa[i];j<n;++j){
            printf("%c",s[j]);
        }  printf("\n");
    }
    printf("\n");
    for(int i=1;i<=n;++i){
        printf("i:%d h:%d\n",i,height[i]);
    }
    printf("\n");
}
int main(){
    int t;scanf("%d",&t);
    while(t--){
        scanf("%s",s);
        n=(int)strlen(s);
        s[n]='#';
        s[++n]='\0';
        da(n+1,255);
        calcheight();
       
//        print();
        
        int ans=0;
        for(int i=2;i<=n;++i){
            int tmp=(n-sa[i]-1)-height[i];
//            printf("%d %d %d \n",sa[i],height[i],tmp);
            ans+=tmp;
        }
        printf("%d\n",ans);
    }
    return 0;
}