SPOJ694&SPOJ705（后缀数组）

这俩题一样。。就后者数据范围大些。

题目：给定一个字符串，求不相同的子串的个数。

每个子串一定是某个后缀的前缀，那么原问题等价于求所有后缀之间的不相
同的前缀的个数。从1到n扫一遍，每一个suffix[sa[i]]的贡献为 n-sa[i]+1- height[i].(sa[i]的前缀个数-重复个数)

#include <cstdio>#include <cstring>#define N 50010int n,m,rank[N<<1],rank1[N],sa[N],h[N],count[N],tmp[N];char s[N];int main(){//freopen("a.in","r",stdin);scanf("%d",&m);while(m--){scanf("%s",s+1);n=strlen(s+1);int ans=0;memset(rank,0,sizeof(rank));memset(count,0,sizeof(count));for(int i=1;i<=n;++i) count[s[i]]=1;for(int i=1;i<=200;++i) count[i]+=count[i-1];for(int i=n;i>=1;--i) rank[i]=count[s[i]];int k=0;for(int p=1;k!=n;p<<=1){memset(count,0,sizeof(count));for(int i=1;i<=n;++i) count[rank[i+p]]++;for(int i=1;i<=n;++i) count[i]+=count[i-1];for(int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;memset(count,0,sizeof(count));for(int i=1;i<=n;++i) count[rank[tmp[i]]]++;for(int i=1;i<=n;++i) count[i]+=count[i-1];for(int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];memcpy(rank1,rank,sizeof(rank1));rank[sa[1]]=k=1;for(int i=2;i<=n;++i){if(rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;rank[sa[i]]=k;}}k=0;for(int i=1;i<=n;++i){if(rank[i]==1){h[1]=0;continue;}if(i==1||h[rank[i-1]]<=1) k=0;if(k) --k;while(s[i+k]==s[sa[rank[i]-1]+k]) ++k;h[rank[i]]=k;}for(int i=1;i<=n;++i){ans+=n-sa[i]+1-h[i];}printf("%d\n",ans);}return 0;}