SPOJ 694 不重复子串

来源：互联网发布：中国内审网络培训编辑：程序博客网时间：2024/05/21 10:23

按照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--nvoid 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;}

0 0