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

[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
[m][i][j];
printf("%d\n",ans);

return 0;
}