bzoj1037【ZJOI2008】生日聚会Party

1037: [ZJOI2008]生日聚会Party

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 1828   Solved: 1081
[ 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

对于100%的数据, n , m ≤ 150，k ≤ 20。 

Source

DP好题...看到题解之后整个人是崩溃的...DP真是什么都能做啊！

用f[i][j][x][y]表示到第i个人，总共j个男生，以i结尾的任意长度中男生最多比女生多x，女生最多比男生多y的方案数。

状态转移如下：

如果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]。

#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define mod 12345678
using namespace std;
int n,m,k,ans,f[301][151][21][21];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int main()
{
	n=read();m=read();k=read();
	f[0][0][0][0]=1;
	F(i,0,n+m-1) F(j,0,n) F(x,0,k) F(y,0,k) if (f[i][j][x][y])
	{
		if (x+1<=k&&j+1<=n) (f[i+1][j+1][x+1][max(y-1,0)]+=f[i][j][x][y])%=mod;
		if (y+1<=k&&i+1-j<=m) (f[i+1][j][max(x-1,0)][y+1]+=f[i][j][x][y])%=mod;
	}
	F(i,0,k) F(j,0,k) (ans+=f[n+m][n][i][j])%=mod;
	printf("%d\n",ans);
}