题目大意:
初始时有i个红球(i∈[0,n])和n-i个蓝球。进行m次操作,每次先任意选出一个球,然后再放入一个蓝球和一个红球,再任意选出一个球。问选出的球的序列有多少种可能?n,m∈[1,3000]
首先,我们用i表示当前红球的数量:
不难发现,如果我们将两次取球合并,那么每次操作必然是以下四种:
1、i>=1时,可以取出两个红球,i-=1
2、i>=1时,可以先取出一个红球,再取出一个蓝球,i不变
3、i<=n时,可以先取出一个蓝球,再取出一个红球,i不变
4、i<=n时,可以取出两个蓝球,i+=1
如果我们把红球数量当做纵坐标,操作次数当做横坐标,那么我们选出的序列就可以看做是一条平面直角坐标系中的路径。这条路径通过平移得到的其他路径,其实是同一种方案。而且,由于i∈[0,n],所以我们的纵坐标也应该在[0,n]之间。以横坐标为第一维,纵坐标为第二维,通过上面的四种操作进行转移,我们就找到了dp的做法。
为了去除重复情况,我们要求每条路径都必须经过一次最高点(即纵坐标为n)。这样就去除了平移后路径不同但方案相同的情况。则我们用dp[i][j][k]作为状态,i,j表示当前在(i,j)位置,布尔变量k表示是否到达过最高点。这样做的空间复杂度是O(n*m)的,由于转移是O(1),所以总时间复杂度是O(n,m)的。
附代码:
#include<bits/stdc++.h>
#define N 3010
using namespace std;
const int MOD=1e9+7;
long long dp[N][N][2];
int n,m;
int ans;
int main(){
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++) dp[0][i][0]=1;
dp[0][n][1]=1;
for(int i=1;i<=m;i++){
for(int j=0;j<=n;j++){
for(int l=(j==n?1:0);l<2;l++){
if(j==0) dp[i][j][l]=dp[i-1][j][l] +dp[i-1][j+1][l];
else if(j==n) dp[i][j][1]=dp[i-1][j][1] +dp[i-1][j-1][1]+dp[i-1][j-1][0];
else dp[i][j][l]=dp[i-1][j-1][l]+dp[i-1][j+1][l]+dp[i-1][j][l]*2;
if(j==n-1) dp[i][j][l]+=(l*2-1)*dp[i-1][j][0];
dp[i][j][l]%=MOD;
}
ans+=dp[m][j][1],ans%=MOD;
}
}
printf("%d\n",ans);
return 0;
}