HDU 4416 Good Article Good sentence

  求多串的子串并集元素的个数,先用没出现过的不同的字符把多个串拼接,用后缀数组求这个串的不同子串的个数,再减去含有拼接字符的子串的个数。用上述方法求『A、B1、……、BN』中不同子串的个数sumAB和『B1、……、BN』中不同子串的个数sumB,答案就是sumAB-sumB。

//2012-09-26 13:22:35	Accepted	4416	671MS	7752K	2071 B	G++	Aros
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 300000+5, MAXM = 100000+5;
int T, N, L[MAXM];
int len, sa[MAXN], height[MAXN], rank[MAXN], tmp[MAXN], top[MAXN];
int *str, a[MAXN];
char A[MAXM];
void makesa()
{
	int lena = len < 256 ? 256 : len;
	memset(top, 0, lena*sizeof(int));
	for (int i = 0; i < len; i++)
		top[rank[i] = str[i]&(-1)]++;
	for (int i = 1; i < lena; i++)
		top[i] += top[i-1];
	for (int i = 0; i < len ; i++)
		sa[--top[rank[i]]] = i;
	for (int k = 1; k < len; k <<= 1)
	{
		for (int i = 0; i < len; i++)
		{
			int j = sa[i]-k;
			if (j < 0)
				j += len;
			tmp[top[rank[j]]++] = j;
		}
		int j = sa[tmp[0]] = top[0] = 0;
		for (int i = 1; i < len; i++)
		{
			if (rank[tmp[i]] != rank[tmp[i-1]] || rank[tmp[i]+k] != rank[tmp[i-1]+k])
				top[++j] = i;
			sa[tmp[i]] = j;
		}
		memcpy(rank, sa , len*sizeof(int));
		memcpy(sa , tmp, len*sizeof(int));
		if (j+1 >= len)
			break;
	}
}
void lcp()
{
	height[0] = 0;
	for (int i = 0, k = 0, j = rank[0]; i+1 < len; i++, k++)
		while (k >= 0 && str[i] != str[sa[j-1]+k])
		{
			height[j] = k--;
			j = rank[sa[j]+1];
		}
}
int main()
{
	scanf("%d", &T);
	for (int cas = 1; cas <= T; cas++)
	{
		scanf("%d%s", &N, A);
		len = 0;
		for (L[0] = 0; A[L[0]]; L[0]++)
			a[len++] = A[L[0]]-'a'+1;
		for (int i = 1; i <= N; i++)
		{
			a[len++] = 26+i;
			scanf("%s", A);
			for (L[i] = 0; A[L[i]]; L[i]++)
				a[len++] = A[L[i]]-'a'+1;
		}
		a[len++] = 0;
		long long sumAB = 0, sumB = 0;
		str = a;
		makesa(); lcp();
		for (int i = 1; i < len; i++)
			sumAB += len-1-sa[i]-height[i];
		long long l = len-1;
		for (int i = 0; i < N; i++)
		{
			l -= L[i];
			sumAB -= (L[i]+1)*l;
			l--;
		}
		str = a+L[0]+1; len -= L[0]+1;
		makesa(); lcp();
		for (int i = 1; i < len; i++)
			sumB += len-1-sa[i]-height[i];
		l = len-1;
		for (int i = 1; i < N; i++)
		{
			l -= L[i];
			sumB -= (L[i]+1)*l;
			l--;
		}
		printf("Case %d: %I64d\n", cas, sumAB-sumB);
	}
	return 0;
}


评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值