UOJ #219 [NOI2016 D1T1] 优秀的拆分 [95分]

如果一个字符串可以被拆分为  AABB AABB 的形式，其中  A A 和  B B 是任意非空字符串，则我们称该字符串的这种拆分是优秀的。

例如，对于字符串 aabaabaa，如果令  A=aab A=aab， B=a B=a，我们就找到了这个字符串拆分成  AABB AABB 的一种方式。

一个字符串可能没有优秀的拆分，也可能存在不止一种优秀的拆分。比如我们令  A=a A=a， B=baa B=baa，也可以用  AABB AABB 表示出上述字符串；但是，字符串 abaabaa 就没有优秀的拆分。

现在给出一个长度为  n n 的字符串  S S，我们需要求出，在它所有子串的所有拆分方式中，优秀拆分的总个数。这里的子串是指字符串中连续的一段。

以下事项需要注意：

1. 出现在不同位置的相同子串，我们认为是不同的子串，它们的优秀拆分均会被记入答案。
2. 在一个拆分中，允许出现  A=B A=B。例如 cccc 存在拆分  A=B=c A=B=c。
3. 字符串本身也是它的一个子串。

输入格式

每个输入文件包含多组数据。输入文件的第一行只有一个整数  T T，表示数据的组数。保证  1≤T≤10 1≤T≤10。

接下来  T T 行，每行包含一个仅由英文小写字母构成的字符串  S S，意义如题所述。

输出格式

输出  T T 行，每行包含一个整数，表示字符串  S S 所有子串的所有拆分中，总共有多少个是优秀的拆分。

样例一

input

4
aabbbb
cccccc
aabaabaabaa
bbaabaababaaba

output

3
5
4
7

explanation

我们用  S[i,j] S[i,j] 表示字符串  S S 第  i i 个字符到第  j j 个字符的子串（从  1 1 开始计数）。

第一组数据中，共有  3 3 个子串存在优秀的拆分：

S[1,4]=aabb S[1,4]=aabb，优秀的拆分为  A=a A=a， B=b B=b；

S[3,6]=bbbb S[3,6]=bbbb，优秀的拆分为  A=b A=b， B=b B=b；

S[1,6]=aabbbb S[1,6]=aabbbb，优秀的拆分为  A=a A=a， B=bb B=bb。

而剩下的子串不存在优秀的拆分，所以第一组数据的答案是  3 3。

第二组数据中，有两类，总共  4 4 个子串存在优秀的拆分：

对于子串  S[1,4]=S[2,5]=S[3,6]=cccc S[1,4]=S[2,5]=S[3,6]=cccc，它们优秀的拆分相同，均为  A=c A=c， B=c B=c，但由于这些子串位置不同，因此要计算  3 3 次；

对于子串  S[1,6]=cccccc S[1,6]=cccccc，它优秀的拆分有  2 2 种： A=c A=c， B=cc B=cc 和  A=cc A=cc， B=c B=c，它们是相同子串的不同拆分，也都要计入答案。

所以第二组数据的答案是  3+2=5 3+2=5。

第三组数据中， S[1,8] S[1,8] 和  S[4,11] S[4,11] 各有  2 2 种优秀的拆分，其中  S[1,8] S[1,8] 是问题描述中的例子，所以答案是  2+2=4 2+2=4。

第四组数据中， S[1,4] S[1,4]， S[6,11] S[6,11]， S[7,12] S[7,12]， S[2,11] S[2,11]， S[1,8] S[1,8] 各有  1 1 种优秀的拆分， S[3,14] S[3,14] 有  2 2 种优秀的拆分，所以答案是  5+2=7 5+2=7。

样例二

见样例数据下载。

样例三

见样例数据下载。

限制与约定

对于全部的测试点,保证  1≤T≤10 1≤T≤10。以下对数据的限制均是对于单组输入数据而言的，也就是说同一个测试点下的  T T 组数据均满足限制条件。

我们假定  n n 为字符串  S S 的长度，每个测试点的详细数据范围见下表：

测试点编号	n n	其他约束
1、2	≤300 ≤300	S S中所有字符全部相同
3、4	≤2000 ≤2000
5、6	≤10 ≤10	无
7、8	≤20 ≤20
9、10	≤30 ≤30
11、12	≤50 ≤50
13、14	≤100 ≤100
15	≤200 ≤200
16	≤300 ≤300
17	≤500 ≤500
18	≤1000 ≤1000
19	≤2000 ≤2000
20	≤30000 ≤30000

时间限制： 1.5s 1.5s

空间限制： 512MB 512MB

下载

样例数据下载

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

hash~

hash大法好！

95分O(n^2)做法：我们枚举中间位置i，然后求出它左右的aa的个数，左右相乘加入答案。

代码当然也是95的哼

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long

const int jin=27;

int t,n,f[2001],kk[2001];
ll l,r,now1,now2,ans;
char s[2001];

int main()
{
	scanf("%d",&t);
	kk[0]=1;
	for(int i=1;i<=2000;i++) kk[i]=kk[i-1]*jin;
	while(t--)
	{
		scanf("%s",s+1);ans=0;
		n=strlen(s+1);
		for(int i=1;i<=n;i++) f[i]=f[i-1]*jin+s[i]-'a';
		for(int i=2;i<n-1;i++)
		{
			l=r=0;
			for(int j=i/2;j;j--)
			{
				now1=f[i]-f[i-j]*kk[j];
				now2=f[i-j]-f[i-j*2]*kk[j];
				l+=(now1==now2);
			}
			for(int j=(n-i)/2;j;j--)
			{
				now1=f[i+j]-f[i]*kk[j];
				now2=f[i+j*2]-f[i+j]*kk[j];
				r+=(now1==now2);
			}
			ans+=l*r;
		}
		printf("%lld\n",ans);
	}
	return 0;
}