记录编号	270659	评测结果	AAAAAAAAAA
题目名称	[SPOJ 705] 不同的子串	最终得分	100
用户昵称	Fmuckss	是否通过	通过
代码语言	C++	运行时间	0.025 s
提交时间	2016-06-15 10:01:04	内存使用	1.89 MiB

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 5e4 + 10;
#define is_ch(a) (a <= 'Z' and a >= 'A')

int a[maxn];
char r[maxn];
int sa[maxn], rk[maxn], he[maxn];
int buk[maxn];
int x[maxn],  y[maxn];
int n, m;

#define same(i, j) (y[sa[i - 1]] == y[sa[i]] and y[sa[i - 1] + j] == y[sa[i] + j])

void DA(int len, int top) {

	for(int i = 0; i < top; i++) buk[i] = 0;
	for(int i = 0; i < len; i++) buk[x[i] = a[i]]++;
	for(int i = 1; i < top; i++) buk[i] += buk[i - 1];
	for(int i = len - 1; i >= 0; i--) sa[--buk[x[i]]] = i;
	
	for(int j = 1; j < len; j <<= 1) {
		int p = 0;
		for(int i = len - j; i < len; i++) y[p++] = i;
		for(int i = 0; i < len; i++) if(sa[i] >= j) y[p++] = sa[i] - j;
		
		for(int i = 0; i < top; i++) buk[i] = 0;
		for(int i = 0; i < len; i++) buk[x[y[i]]]++;
		for(int i = 1; i < top; i++) buk[i] += buk[i - 1];
		for(int i = len - 1; i >= 0; i--) sa[--buk[x[y[i]]]] = y[i];
		
		swap(x, y);
		p = 1; x[sa[0]] = 0;
		for(int i = 1; i < len; i++) x[sa[i]] = same(i, j) ? p - 1 : p++;
		if(p >= len) break;
		top = p;
	}
}

int c[maxn];

void get_rank(int len) {
	for(int i = 0; i <= len; i++) rk[sa[i]] = i;
}

void get_height(int len) {
	int k = 0;
	for(int i = 0; i < len; i++) {
		if(k) k--;
		int j = sa[rk[i] - 1];
		while(a[i + k] == a[j + k]) k++;
		he[rk[i]] = k;
	}
}

int get_line(int *tar) {
	char tmp = getchar();
	int i = 0;
	while(!is_ch(tmp)) tmp = getchar();
	while( is_ch(tmp)) {
		m = max(m, tar[i++] = tmp);
		tmp = getchar();
	}
	tar[i] = '$';
	return i;
}

void test() {
	for(int i = 1; i <= n; i++) {
		for(int j = sa[i]; j < n; j++) {
			printf("%c", a[j]);
		}
		printf("\n");
	}
}

void solve() {
	n = get_line(a);
	m++;
	DA(n + 1, m);
//	build_sa(n, m);
//	test();
	get_rank(n);
	get_height(n);
	int tot = 0;
	for(int i = 0; i <= n; i++) {
		tot += n - sa[i] - he[i];
	}
	printf("%d\n", tot);
}


int main() {
	freopen("subst1.in", "r", stdin);
	freopen("subst1.out", "w", stdout);
	solve();
	return 0;
}