【bzoj1037】[ZJOI2008]生日聚会Party

1037: [ZJOI2008]生日聚会Party

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

[ Submit][ Status][ Discuss]


真的觉得自己的DP菜的半死。。
这题的方程比较难想
考虑到，往一个状态后加男生或者是女生，影响这个状态的合法性的只有这个状态的后缀的男生减女生人数的最大值或者是女生减男生人数的最大值（很拗口但是不难理解）
观察到k特别小，于是我们考虑把上述两个因素加入到状态内
设f[i][j][p][q]为当男生人数为i，女生人数为j，后缀的男生人数减女生人数的最大值为p，女生人数减男生人数的最大值为q
那么显然有 f[i][j][p][q] = f[i - 1][j][p - 1][q + 1] + f[i][j - 1][p + 1][q - 1]
边界注意判一下就好了
代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int maxn = 160;
const int maxk = 25;
const int crz = 12345678;

int ans,n,m,k;
int f[maxn][maxn][maxk][maxk];

inline int getint()
{
	int ret = 0;
	char c = getchar();
	while (c < '0' || c > '9') c = getchar();
	while (c >= '0' && c <= '9')
		ret = ret * 10 + c - '0',c = getchar();
	return ret;
}

int main()
{
	n = getint(); m = getint(); k = getint();
	if (!m || !n) {printf(n + m <= k ? "1" : "0"); return 0;}
	f[0][1][1][0] = f[1][0][0][1] = 1;
	for (int i = 0; i <= n; i++)
	{
		for (int j = 0; j <= m; j++)
		{
			if ((i == 0 && j == 1) || (i == 1 && j == 0)) continue;
			for (int p = 0; p <= min(k,j); p++)
				for (int q = 0; q <= min(k,i); q++)
				{
					if (p == 0 && q == 0) continue;
					if (q && i)
					{
						f[i][j][p][q] = (f[i - 1][j][p + 1][q - 1] + f[i][j][p][q]) % crz;
						if (p == 0) f[i][j][p][q] = (f[i - 1][j][0][q - 1] + f[i][j][p][q]) % crz;
					}
					if (p && j)
					{
						f[i][j][p][q] = (f[i][j - 1][p - 1][q + 1] + f[i][j][p][q]) % crz;
						if (q == 0) f[i][j][p][q] = (f[i][j - 1][p - 1][0] + f[i][j][p][q]) % crz;
					}
				}
		}
	}
	for (int i = 0; i <= k; i++)
		for (int j = 0; j <= k; j++)
			ans = (ans + f[n][m][i][j]) % crz;
	printf("%d",ans);
	return 0;
}