Bzoj 1566: [NOI2009]管道取珠(DP)

1566: [NOI2009]管道取珠

Time Limit: 20 Sec Memory Limit: 650 MB

Submit: 1558 Solved: 890

[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；

/*
一道DP.
比较妙.
题意相同类型的序列有x个,对答案贡献是x^2。
等价于两个人各自进行操作,得到相同类型序列的方案数.
这个东西是比较好证的。
我们设一个合法序列状态是A,并且现在有x,y两种取法均能得到A
根据乘法原理,那么问题就转化为求A(x,y)二元组的个数.
想象一下现在有两个人正在取数,要求两个人取数相同的方案个数.
令f[i][j][k]表示
people 1 在上面取了i个,下面取了j个
people 2 在上面取了k个，下面取了i+j-k个
people1和people2都取了i+j个数的相同方案的方案个数。
然后枚举上一次取的位置转移即可。。
*/
#include<iostream>
#include<cstdio>
#define mod 1024523
#define MAXN 501
using namespace std;
char a[MAXN],b[MAXN];
int n,m,f[MAXN][MAXN][MAXN];
int main()
{
    scanf("%d %d",&n,&m);
    scanf("%s",a+1);
    scanf("%s",b+1);
    for(int i=1;i<=n/2;i++) swap(a[i],a[n-i+1]);
    for(int i=1;i<=m/2;i++) swap(b[i],b[m-i+1]);
    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++)
        {
            int l=i+j-k;int * p=&f[i][j][k];
            if(l<0||l>m) continue;
            if(i&&k&&a[i]==a[k]) *p=(*p+f[i-1][j][k-1])%mod;
            if(i&&l&&a[i]==b[l]) *p=(*p+f[i-1][j][k])%mod;
            if(j&&k&&b[j]==a[k]) *p=(*p+f[i][j-1][k-1])%mod;
            if(j&&l&&b[j]==b[l]) *p=(*p+f[i][j-1][k])%mod;
        }
    printf("%d",f[n][m][n]);
    return 0;
}