【SPOJ】694. Distinct Substrings

http://www.spoj.com/problems/DISUBSTR/

题意：求字符串不同子串的数目。

#include <bits/stdc++.h>

using namespace std;

const int N=1005;

void sort(int *x, int *y, int *sa, int n, int m) {

	static int c[N], i;

	for(i=0; i<m; ++i) c[i]=0;

	for(i=0; i<n; ++i) ++c[x[y[i]]];

	for(i=1; i<m; ++i) c[i]+=c[i-1];

	for(i=n-1; i>=0; --i) sa[--c[x[y[i]]]]=y[i];

}

void hz(int *a, int *sa, int n, int m) {

	static int t1[N], t2[N], i, j, p, *x, *y, *t;

	x=t1, y=t2;

	for(i=0; i<n; ++i) x[i]=a[i], y[i]=i;

	sort(x, y, sa, n, m);

	for(j=1, p=1; p<n; j<<=1, m=p) {

		p=0;

		for(i=n-j; i<n; ++i) y[p++]=i;

		for(i=0; i<n; ++i) if(sa[i]-j>=0) y[p++]=sa[i]-j;

		sort(x, y, sa, n, m);

		for(t=x, x=y, y=t, p=1, x[sa[0]]=0, i=1; i<n; ++i)

			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;

	}

}

void geth(int *s, int *sa, int *h, int n) {

	static int rank[N], j, i, k;

	for(i=1; i<=n; ++i) rank[sa[i]]=i;

	for(k=0, i=1; i<=n; h[rank[i++]]=k)

		for(k?--k:0, j=sa[rank[i]-1]; s[i+k]==s[j+k]; ++k);

}

int a[N], sa[N], h[N], n;

char s[N];

int main() {

	int cs;

	scanf("%d", &cs);

	while(cs--) {

		scanf("%s", s+1);

		n=strlen(s+1);

		for(int i=1; i<=n; ++i) a[i]=s[i];

		hz(a, sa, n+1, 128);

		geth(a, sa, h, n);

		int ans=0;

		for(int i=1; i<=n; ++i) ans+=n-sa[i]+1-h[i];

		printf("%d\n", ans);

	}

	return 0;

}

　　

---

经典题....首先每个后缀的前缀就是一个子串，因此每个后缀可以构成这个后缀长度大小那么多个子串。但是我们要考虑重合的情况，即我们剪掉与上一个后缀子串相同前缀的height值就好啦