NOI2009管道取珠(dp)

时间:2023-03-10 01:41:32
NOI2009管道取珠(dp)

题意:给定两列球,可以从任意一列球的末尾弹出一个球,最后会得到一个序列,设第i种序列可以被a[i]种操作产生,那么会产生a[i]^2的贡献,求贡献和、

Solution:

首先我们观察a[i]^2的含义,发现它是有a[i]种序列两两之间产生1的贡献。

于是我们就有了一个dp的思路,dp[i][j][k][l]表示一种序列为在第一列有i个,另一列有j个,另一种序列在第一列有k个,在第二列有l个。他们产生一样的输出序列的方案数。

转移就枚举一下一个弹什么。

因为i+j=k+l所以我们可以去掉一维,第一位数组也可以滚动,空间复杂度n^2,时间复杂度n^3.

Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 503
using namespace std;
const int mod=;
int dp[N][N][N],n,m,now;
char s1[N],s2[N];
inline void mode(int &x){
while(x>=mod)x-=mod;
}
int main(){
scanf("%d%d",&n,&m);
scanf("%s%s",s1+,s2+);
reverse(s1+,s1+n+);reverse(s2+,s2+m+);
dp[][][]=;now=;
for(int i=;i<=n;++i,now^=){
memset(dp[now^],,sizeof(dp[now^]));
for(int j=;j<=m;++j)
for(int k=;k<=n;++k)if(dp[now][j][k]){
int l=i+j-k,num=dp[now][j][k];
if(l<||l>m)continue;
if(s1[i+]==s1[k+])mode(dp[now^][j][k+]+=num);
if(s1[i+]==s2[l+])mode(dp[now^][j][k]+=num);
if(s2[j+]==s1[k+])mode(dp[now][j+][k+]+=num);
if(s2[j+]==s2[l+])mode(dp[now][j+][k]+=num);
}
}
cout<<dp[now][m][n];
return ;
}