SPOJ 705

来源：互联网发布：数据透视表刷新数据编辑：程序博客网时间：2024/06/04 00:53

题目大意：求一个字符串中本质不同的串的多少
思路：显然任意一个字串都可以看成一个后缀的前缀。那么对于一个新加入的后缀sa[i]来说，它能够产生的贡献是w[i]=n-sa[i]+1,但其中height[i]的前缀是与sa[i-1]重复，所以
w[i]=n-sa[i]+1-h[i] ans=n*n(n-1)/2-h[i](1 to n)
PS:这题字母是大写

#include<iostream>#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fod(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=1e3+10;int n,p,q,k;char ch[N];int a[N],v[N],h[N];int sa[2][N],rk[2][N];void calsa(int sa[N],int rk[N],int SA[N],int RK[N]){    for(int i=1;i<=n;i++)v[rk[sa[i]]]=i;    for(int i=n;i;i--)        if(sa[i]>k)            SA[v[rk[sa[i]-k]]--]=sa[i]-k;    for(int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;    for(int i=1;i<=n;i++)        RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);}void getsa(){    memset(v,0,sizeof(v));    p=0,q=1;    for(int i=1;i<=n;i++)v[a[i]]++;    for(int i=1;i<=30;i++)v[i]+=v[i-1];    for(int i=1;i<=n;i++)        sa[p][v[a[i]]--]=i;    for(int i=1;i<=n;i++)        rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);    for(k=1;k<n;k<<=1,swap(p,q))        calsa(sa[p],rk[p],sa[q],rk[q]);}void geth(){    k=0;    for(int i=1;i<=n;i++)        if(rk[p][i]==1)h[rk[p][i]]=0;        else         {            int j=sa[p][rk[p][i]-1];            while(a[i+k]==a[j+k])k++;            h[rk[p][i]]=k;if(k>0)k--;        }}void solve(){    int ans=0;        fo(i,1,n) {            ans+=(n-sa[p][i]-h[i]+1);        }    //  fo(i,1,n) printf("sa[p][%d]=%d\n",i,sa[p][i]);    cout<<ans<<endl;}int main(){    int T;    scanf("%d",&T);    while(T--){        scanf("%s",ch+1);        n=strlen(ch+1);        fo(i,1,n) a[i]=ch[i]-'A'+1;        getsa(); geth();        solve();    }    return 0;}

阅读全文

0 0