BZOJ 1037-生日聚会Party（DP）

1037: [ZJOI2008]生日聚会Party

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 2519   Solved: 1509
[ Submit][ Status][ Discuss]

Description

　　今天是hidadz小朋友的生日，她邀请了许多朋友来参加她的生日party。 hidadz带着朋友们来到花园中，打算
坐成一排玩游戏。为了游戏不至于无聊，就座的方案应满足如下条件：对于任意连续的一段，男孩与女孩的数目之
差不超过k。很快，小朋友便找到了一种方案坐了下来开始游戏。hidadz的好朋友Susie发现，这样的就座方案其实
是很多的，所以大家很快就找到了一种，那么到底有多少种呢？热爱数学的hidadz和她的朋友们开始思考这个问题
…… 假设参加party的人中共有n个男孩与m个女孩，你是否能解答Susie和hidadz的疑问呢？由于这个数目可能很
多，他们只想知道这个数目除以12345678的余数。

Input

　　仅包含一行共3个整数，分别为男孩数目n,女孩数目m,常数k。

Output

　　应包含一行，为题中要求的答案。

Sample Input

1 2 1

Sample Output

1

HINT

n , m ≤ 150，k ≤ 20。

Source

DP好题，定义DP【i】【j】【num1】【num2】：前i个人中，共有j个男生，

且以i结尾的任意长度的男生最多比女生多num1，最多比女生少num2的方案数。

状态转移：

如果x+1<=k，f[i+1,j+1,x+1,max(y-1,0)]+=f[i][j][x][y]

如果y+1<=k，f[i+1,j,max(x-1,0),y+1]+=f[i][j][x][y]

最终答案为∑(0≤i≤k)∑(0≤j≤k)f[n+m][n][i][j]。

不得不说，DP是真的厉害。。

#include<map>  
#include<stack>  
#include<queue>  
#include<vector>  
#include<math.h>  
#include<stdio.h>  
#include<iostream>  
#include<string.h>  
#include<stdlib.h>  
#include<algorithm>  
using namespace std;  
typedef long long  ll;  
#define inf 1000000000  
#define mod 12345678  
#define  maxn  2005
#define  lowbit(x) (x&-x)  
#define  eps 1e-10  
int dp[305][155][25][25];
int  main(void)
{
	int n,m,k,i,j,num1,num2,ans=0;
	scanf("%d%d%d",&n,&m,&k);
	dp[0][0][0][0]=1;
	for(i=0;i<n+m;i++)
		for(j=0;j<=n;j++)
			for(num1=0;num1<=k;num1++)
				for(num2=0;num2<=k;num2++)
					if(dp[i][j][num1][num2])
					{
						if(num1+1<=k)
						{
							dp[i+1][j+1][num1+1][max(num2-1,0)]+=dp[i][j][num1][num2];
							dp[i+1][j+1][num1+1][max(num2-1,0)]%=mod;
						}
						if(num2+1<=k)
						{
							dp[i+1][j][max(num1-1,0)][num2+1]+=dp[i][j][num1][num2];
							dp[i+1][j][max(num1-1,0)][num2+1]%=mod;
						}
					}
	for(i=0;i<=k;i++)
		for(j=0;j<=k;j++)
			ans=(ans+dp[n+m][n][i][j])%mod;
	printf("%d\n",ans);
	return 0;
}