这俩题一样。。就后者数据范围大些。
题目:给定一个字符串,求不相同的子串的个数。
每个子串一定是某个后缀的前缀,那么原问题等价于求所有后缀之间的不相
同的前缀的个数。从1到n扫一遍,每一个suffix[sa[i]]的贡献为 n-sa[i]+1- height[i].(sa[i]的前缀个数-重复个数)
#include <cstdio>
#include <cstring>
#define N 50010
int 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;
}