【SPOJ 694】Distinct Substrings

【链接】h在这里写链接

【题意】

    给你一个长度最多为1000的字符串

    让你求出一个数x,这个x=这个字符串的不同子串个数;

【题解】

    后缀数组题。
    把原串复制一份,加在原串后面(中间用分隔符分开),这样每个子串都能重复出现一次了,枚举Height的时候就都能枚举到了.

    求出来它的后缀数组以及Height值。

    然后枚举子串的长度为2..1000->x;
    对于连续的一块Height>=x的,答案++.
    长度为1的特判一下就好.
    (能够照顾到所有的子串)

    输出结果即可

【错的次数】

0

【反思】

在这了写反思

【代码】

#include<bits/stdc++.h>
using namespace std;

const int N = 2e3;
const int MAX_CHAR = 300;//每个数字的最大值。
char s[N + 10];//如果是数字,就写成int s[N+10]就好,从0开始存
int Sa[N + 10], T1[N + 10], T2[N + 10], C[N + 10];
int Height[N + 10], Rank[N + 10];

void build_Sa(int n, int m) {
	int i, *x = T1, *y = T2;
	for (i = 0; i<m; i++) C[i] = 0;
	for (i = 0; i<n; i++) C[x[i] = s[i]]++;
	for (i = 1; i<m; i++) C[i] += C[i - 1];
	for (i = n - 1; i >= 0; i--) Sa[--C[x[i]]] = i;
	for (int k = 1; k <= n; k <<= 1)
	{
		int p = 0;
		for (i = n - k; i<n; i++) y[p++] = i;
		for (i = 0; i<n; i++) if (Sa[i] >= k) y[p++] = Sa[i] - k;
		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];
		swap(x, y);
		p = 1; x[Sa[0]] = 0;
		for (i = 1; i<n; i++)
			x[Sa[i]] = y[Sa[i - 1]] == y[Sa[i]] && y[Sa[i - 1] + k] == y[Sa[i] + k] ? p - 1 : p++;
		if (p >= n) break;
		m = p;
	}
}

void getHeight(int n)
{
	int i, j, k = 0;
	for (i = 1; i <= n; i++) Rank[Sa[i]] = i;
	for (i = 0; i<n; i++) {
		if (k) k--;
		j = Sa[Rank[i] - 1];
		while (s[i + k] == s[j + k]) k++;
		Height[Rank[i]] = k;
	}
}

int bo[300];

int main() {
	//freopen("F:\rush.txt", "r", stdin);
	int T;
	scanf("%d", &T);
	while (T--)
	{
		scanf("%s", s);
		int n = strlen(s),tn = n;
		s[n++] = 'z' + 1;
		for (int i = 1; i <= tn; i++)
			s[n++] = s[i - 1];
		s[n] = 0;
		build_Sa(n + 1, MAX_CHAR);//注意调用n+1
		getHeight(n);
		int ans = 0;
		for (int i = 2;i <= tn;i++)
			for (int j =2;j <= n;j++)
				if (Height[j] >= i)
				{
					int k = j;
					while (k + 1 <= n && Height[k + 1] >= i) k++;
					ans++;
					j = k;
				}
		memset(bo, 0, sizeof bo);
		for (int i = 0; i < tn; i++)
			if (bo[s[i]] == 0) 
			{
				bo[s[i]] = 1;
				ans++;
			}
		printf("%d
", ans);
			
	}
	return 0;
}