[JZOJ 1598]文件修复

Description

　　有一个文件被破坏了，可是值得庆幸的是，只是文件的顺序被打乱了。文件仅包含大小写的拉丁字母以及逗号，句号和叹号。为了尽快修复，请你找出有多少个至少出现两次的子串。
　　比如字符串abbabc，子串”a”，”b”,”ab”分别出现了2次，3次，2次。

Input

　　输入文件第一行包含一个整数n表示文件的长度。
　　第二行n个字符，表示被破坏的文件。

Output

　　输出一个数，表示有多少个至少出现两次的子串。

Sample Input

aabbabb

Sample Output

5

Data Constraint

Hint

【数据约束和评分方法】
　　对于10%的数据，1<=n<=100。
　　对于30%的数据，1<=n<=1000。
　　对于50%的数据，1<=n<=10000。　　
　　对于100%的数据，1<=n<=100000。

方法什么的挺好想的，用sa求出height之后，假如height[i] - height[i-1] > 0的话，就将差值累加进ans就可以了。

那么为什么要减去前一个的height呢，因为我们对于同一个串，我们只算一遍，所以要减去之前算过的

#include<cstdio>
#include<algorithm>
#define maxn 100100
using namespace std;
char ch;
int ans,tot,n,ws[maxn],x[maxn],y[maxn],wv[maxn],a[maxn],h[maxn],sa[maxn],t[maxn];
bool get(int x){
	if (x >= 'a' && x <= 'z') return true;
	if (x >= 'A' && x <= 'Z') return true;
	if (x == ',' || x == '.' || x == '!') return true;
	return false;
}

void init(){
	while (ch = getchar(),(! get(ch)));
	a[++ tot] = ch;
	while (ch = getchar(),(get(ch))) a[++ tot] = ch;
	n = tot;
}

bool cmp(int *rank,int a,int b,int l){
	return (rank[a] == rank[b] && rank[a+l] == rank[b+l]);
}

void da(int *x,int *y,int *t){
	int i,j,p,m = 122;
	for (i = 1 ; i <= m ; i ++) ws[i] = 0;
	for (i = 1 ; i <= n ; i ++) ws[x[i] = a[i]] ++;
	for (i = 2 ; i <= m ; i ++) ws[i] += ws[i - 1];
	for (i = n ; i >= 1 ; i --) sa[ws[x[i]] --] = i;
	for (j = 1,p = 1 ; p <= n ; j *= 2,m = p){
		for (p = 0,i = n - j + 1 ; i <= n ; i ++) y[++ p] = i;//!!
		for (i = 1 ; i <= n ; i ++) if (sa[i] > j) y[++ p] = sa[i] - j;//!!
		for (i = 1 ; i <= n ; i ++) wv[i] = x[y[i]];
		for (i = 1 ; i <= m ; i ++) ws[i] = 0;
		for (i = 1 ; i <= n ; i ++) ws[wv[i]] ++;
		for (i = 1 ; i <= m ; i ++) ws[i] += ws[i - 1];
		for (i = n ; i >= 1 ; i --) sa[ws[wv[i]] --] = y[i];
		for (t = x,x = y,y = t,p = 2,x[sa[1]] = 1,i = 2 ; i <= n ; i ++)
			x[sa[i]] = cmp(y , sa[i - 1] , sa[i] , j) ? p - 1 : p ++;
	}
}

void calc(){
	int k = 0,j,i;
	for (i = 1 ; i <= n ; i ++) x[sa[i]] = i;
	for (i = 1 ; i <= n ; h[x[i ++]] = k)
		for (k ? k -- : 0,j = sa[x[i] - 1] ; a[i + k] == a[j + k] ; k ++);
}

void work(){
	da(x,y,t);
	calc();
	for (int i = 1 ; i <= n ; i++) ans += max(h[i] - h[i - 1] , 0);
}

int main(){
	init();
	work();
	printf("%d",ans);
}