1566: [NOI2009]管道取珠
Time Limit: 20 Sec Memory Limit: 650 MB
Submit: 1277 Solved: 721
[Submit][Status][Discuss]
Description
Input
第一行包含两个整数n, m,分别表示上下两个管道中球的数目。 第二行为一个AB字符串,长度为n,表示上管道中从左到右球的类型。其中A表示浅色球,B表示深色球。 第三行为一个AB字符串,长度为m,表示下管道中的情形。
Output
仅包含一行,即为 Sigma(Ai^2) i从1到k 除以1024523的余数。
Sample Input
2 1
AB
B
Sample Output
5
HINT
样例即为文中(图3)。共有两种不同的输出序列形式,序列BAB有1种产生方式,而序列BBA有2种产生方式,因此答案为5。
【大致数据规模】
约30%的数据满足 n, m ≤ 12;
约100%的数据满足n, m ≤ 500。
题目中给的平方是有用的。
对于其中的一个
ai
来说,相当于每一个都要跟其他的算一遍,这样就得到了平方。
这也就相当于操作两次,所得到的颜色序列一样的方案数。这个东西就可以直接
dp
了。
设
f[i][j][p][q]
表示第一次上面选了
i
个,下面选了
因为
i+j=p+q
所以去掉最后一维。
复杂度:
n3
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define Mod 1024523
const int N=510;
int n,m,f[2][N][N],a[N],b[N];
inline void add(int &x,int y){
x+=y;
if(x>Mod) x-=Mod;
}
int main(){
int i,j,k,now=0;
scanf("%d%d",&n,&m);
for(i=1;i<=n;++i){
char ch=getchar();
while(ch!='A'&&ch!='B') ch=getchar();
a[i]=ch-'A'+1;
}
for(i=1;i<=m;++i){
char ch=getchar();
while(ch!='A'&&ch!='B') ch=getchar();
b[i]=ch-'A'+1;
}
f[now][0][0]=1;
for(i=0;i<=n;++i){
for(j=0;j<=m;++j)
for(k=0;k<=n;++k)
f[now^1][j][k]=0;
for(j=0;j<=m;++j)
for(k=0;k<=n;++k){
if(a[i+1]==a[k+1]) add(f[now^1][j][k+1],f[now][j][k]);
if(a[i+1]==b[i+j+1-k]) add(f[now^1][j][k],f[now][j][k]);
if(b[j+1]==a[k+1]) add(f[now][j+1][k+1],f[now][j][k]);
if(b[j+1]==b[i+j+1-k]) add(f[now][j+1][k],f[now][j][k]);
}
now^=1;
}
printf("%d\n",f[now][m][n]);
}