Atcoder Grand Contest 013D piling Up

题目大意：

初始时有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;
}