BZOJ.1037.[ZJOI2008]生日聚会Party(DP)

题目链接

f[i][j][a][b]，表示当前已有i个男生、j个女生，前面这一段(更前面的保证满足)男生至多比女生多a人，女生最多比男生多b人。
如果当前选择男生，f[i+1][j][a+1][b-1]+=f[i][j][a][b]；否则f[i][j+1][a-1][b+1]+=f[i][j][a][b]。注意一下边界。

//44504kb	156ms
#include <cstdio>
#include <algorithm>
#define mod (12345678)
#define Mod(x) (x>=mod)&&(x-=mod)
const int N=152;

int main()
{
	static int f[N][N][22][22];
	int n,m,K; scanf("%d%d%d",&n,&m,&K);
	f[0][0][0][0]=1;
	for(int i=0; i<=n; ++i)//要到n。
		for(int j=0; j<=m; ++j)
			for(int a=0; a<=K; ++a)
				for(int v,b=0; b<=K; ++b)
					if(v=f[i][j][a][b])
					{
						int &boy=f[i+1][j][a+1][std::max(0,b-1)], &girl=f[i][j+1][std::max(0,a-1)][b+1];
						boy+=v, Mod(boy);
						girl+=v, Mod(girl);
					}
	int ans=0;
	for(int i=0; i<=K; ++i,ans%=mod)
		for(int j=0; j<=K; ++j) ans+=f[n][m][i][j];
	printf("%d\n",ans);

	return 0;
}