[NOI2009]管道取珠

Posted by Dispwnl on April 1, 2018

题目

题目描述

管道取珠是小X很喜欢的一款游戏。在本题中,我们将考虑该游戏的一个简单改版。游戏画面如图1所示:

游戏初始时,左侧上下两个管道分别有一定数量的小球(有深色球和浅色球两种类型),而右侧输出管道为空。每一次操作,可以从左侧选择一个管道,并将该管道中最右侧的球推入右边输出管道。

例如:我们首先从下管道中移一个球到输出管道中,将得到图2所示的情况。

假设上管道中有$n$个球, 下管道中有$m$个球,则整个游戏过程需要进行$n+m$次操作,即将所有左侧管道中的球移入输出管道。最终$n+m$个球在输出管道中从右到左形成输出序列。

爱好数学的小X知道,他共有C(n+m,n)种不同的操作方式,而不同的操作方式可能导致相同的输出序列。举个例子,对于图3所示的游戏情形:

我们用A表示浅色球,B表示深色球。并设移动上管道右侧球的操作为U,移动下管道右侧球的操作为D,则共有C(2+1,1)=3种不同的操作方式,分别为UUD,UDU,DUU;最终在输出管道中形成的输出序列(从右到左)分别为BAB,BBA,BBA。可以发现后两种操作方式将得到同样的输出序列。

假设最终可能产生的不同种类的输出序列共有K种,其中:第i种输出序列的产生方式(即不同的操作方式数目)有ai个。聪明的小X早已知道,

Σai=C(n+m,n)

因此,小X希望计算得到:

Σ(ai)^2

你能帮助他计算这个值么?由于这个值可能很大,因此只需要输出该值对1024523的取模即可(即除以1024523的余数)。

说明:文中C(n+m,n)表示组合数。组合数C(a,b)等价于在a个不同的物品中选取b个的选取方案数。

输入输出格式

输入格式:

输入文件中的第一行为两个整数n,m,分别表示上下两个管道中球的数目。

第二行中为一个AB字符串,长度为n,表示上管道中从左到右球的类型。其中:A表示浅色球,B表示深色球。

第三行中为一个AB字符串,长度为m,表示下管道中的情形。

输出格式:

输出文件中仅一行为一个整数,即为 除以1024523的余数。

输入输出样例

输入样例#1:

2 1
AB
B

输出样例#1:

5

说明

【样例说明】

样例即为文中(图3)。共有两种不同的输出序列形式,序列BAB有1种产生方式,而序列BBA有2种产生方式,因此答案为5。

【数据规模和约定】

对于30%的数据,满足:m,n<=12;

对于100%的数据,满足:m,n<=500。

题解

恶心好题

首先$\sum_{1}^{k}{ai}^{2}$这个玩意得转换一下

怎么转换呢?

根据题目中的定义,$ai$是第$i$种输出序列的产生方式(即不同的操作方式数目)

那么我们能把${ai}^{2}$看成是两个人玩这个游戏,两个人取出第i种输出序列的方法的组合(乘法原理)

然后考虑$dp$

既然是两个人,$f_{i,j,k,l}​$表示第一个人第一个管道取前$i​$个,第二个管道取前$j​$个,第二个人第一个管道取前$k​$个,第二个管道取前$l​$个

然后你会发现$n,m\le 500$,空间炸了

为了使两个人取出的序列相同,前提是取得一样多,所以$i+j=k+l$

所以可以优化掉一维

然而在bzoj上这样已经可以过了,但$luogu$上空间还是炸

发现$f_i$是由$f_i$或$f_{i-1}$递推过来的

所以可以用滚动数组优化空间到$n^2​$

代码

# include<iostream>
# include<cstring>
# include<cstdio>
# define LL long long
using namespace std;
const int MAX=505;//这里要开大点不然会RE
const LL mod=1024523;
int n,m;
string a,b;
LL f[2][MAX][MAX];
bool fl;
int read()
{
	int x=0;
	char ch=getchar();
	for(;!isdigit(ch);ch=getchar());
	for(;isdigit(ch);x=x*10+ch-48,ch=getchar());
	return x;
}
string Read()
{
	string a="";
	char ch=getchar();
	for(;ch!='B'&&ch!='A';ch=getchar());
	for(;ch=='B'||ch=='A';a+=ch,ch=getchar());
	return a;
}
int main()
{
	n=read(),m=read(),a=Read(),b=Read();
	f[0][0][0]=1;
	for(int i=0;i<=n;++i)
	  {
	  	for(int j=0;j<=m;++j)
	      for(int k=0;k<=n;++k)
	      	{
	      		if(i+j-k>m||i+j-k<0) continue;
	      		if(a[i]==a[k]) f[fl^1][j][k+1]=(f[fl^1][j][k+1]+f[fl][j][k])%mod;
	      		if(b[j]==b[i+j-k]) f[fl][j+1][k]=(f[fl][j+1][k]+f[fl][j][k])%mod;
	      		if(a[i]==b[i+j-k]) f[fl^1][j][k]=(f[fl^1][j][k]+f[fl][j][k])%mod;
	      		if(a[k]==b[j]) f[fl][j+1][k+1]=(f[fl][j+1][k+1]+f[fl][j][k])%mod;
	      		f[fl][j][k]=0;
		  	}
		fl^=1;
	  }
	printf("%lld",f[fl][m][n]);
	return 0;
}