SPOJ694&SPOJ705（后缀数组）

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

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

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

#include 
#include 
#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;
}