SPOJ 694 Distinct Substrings 后缀数组

Given a string, we need to find the total number of its distinct substrings.

Input

T- number of test cases. T<=20;
Each test case consists of one string, whose length is <= 1000

Output

For each test case output one number saying the number of distinct substrings.

Example

Sample Input:
2
CCCCC
ABABA

Sample Output:
5
9

Explanation for the testcase with string ABABA:
len=1 : A,B
len=2 : AB,BA
len=3 : ABA,BAB
len=4 : ABAB,BABA
len=5 : ABABA
Thus, total number of distinct substrings is 9.

传送门

题意就是给出一个字符串，让你求它的不同的子串的个数。

首先子串，就是某一个后缀的某一个前缀。

我们可以考虑利用容斥原理，计算出重复的子串个数，然后用总子串个数减去就好了。

此题首先用sa来求出所有的height，

我们知道height的意义是两个后缀的最长公共前缀，

而每个后缀都是不同的，所以他们的最长公共前缀的部分是重复的。

举个例子，两个后缀的最长公共前缀是x，

那么它们的第一位，第二位……第x位的子串重复了。

aaaaaaa

 aaaaaaaa

这是上面2个后缀的最长公共前缀部分，那么显然，

a，aa，aaa……都是重复的子串。

所以我们只要求出所有height的和sum，再用总子串的个数减掉就好了。

注意一下多组数据，rank要清零……不然就是RE..

#include<bits/stdc++.h>
using namespace std;
const int 
	N=1005;
int len,ans1;
char s[N];
int cnta[N],cntb[N],a[N],b[N];
int tsa[N],sa[N],rank[N<<1];
void Get_SA(){
	for (int i=0;i<=300;i++) cnta[i]=0;
	for (int i=1;i<=len;i++) cnta[s[i]]++;
	for (int i=1;i<=300;i++) cnta[i]+=cnta[i-1];
	for (int i=len;i;i--) sa[cnta[s[i]]--]=i;
	rank[sa[1]]=1;
	for (int i=2;i<=len;i++)
		rank[sa[i]]=rank[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);
	for (int j=1;rank[sa[len]]!=len;j<<=1){
		for (int i=1;i<=len;i++) a[i]=rank[i],b[i]=rank[i+j];
		for (int i=0;i<=len;i++) cnta[i]=cntb[i]=0;
		for (int i=1;i<=len;i++) cnta[a[i]]++,cntb[b[i]]++;
		for (int i=1;i<=len;i++) cnta[i]+=cnta[i-1],cntb[i]+=cntb[i-1];
		for (int i=len;i;i--) tsa[cntb[b[i]]--]=i;
		for (int i=len;i;i--) sa[cnta[a[tsa[i]]]--]=tsa[i];
		rank[sa[1]]=1;
		for (int i=2;i<=len;i++)
			rank[sa[i]]=rank[sa[i-1]]+
				(a[sa[i]]!=a[sa[i-1]] || b[sa[i]]!=b[sa[i-1]]);
	}
}
void Get_H(){
	int t=0;
	for (int i=1;i<=len;i++){
		if (t) t--;
		while (s[i+t]==s[sa[rank[i]-1]+t]) t++;
	//	height[i]=len;
		ans1+=t;
	}
}
void solve(){
	memset(rank,0,sizeof(rank));
	
	len=strlen(s+1);
	int ans=(len*(len+1))>>1;
	ans1=0;

	Get_SA(),Get_H();

	printf("%d\n",ans-ans1);
} 
int main(){
	int n;scanf("%d",&n);
	while (n--){
		scanf("%s",s+1);
		solve();
	}
	return 0;
}