题解 BZOJ 1037 & Luogu P2592 [ZJOI2008]生日聚会

BZOJ & Luogu

---

老师说是背包？并没看出来QAQ

设f[i][j][o][p]表示已经选了i个人，j个男生，男生比女生最多多o个，女生比男生最多多p个时的方案数

两种转移：

if(j+<=n&&o+<=k) (f[i+][j+][o+][max(p-,)]+=f[i][j][o][p])%=mod;
if(i+-j<=m&&p+<=k) (f[i+][j][max(o-,)][p+]+=f[i][j][o][p])%=mod;

第一种是说：如果男生人数+1小于n且新的差值（就是加了一个男生后）<=k，就可以转移；

第二种是说：如果女生人数（女生人数=总人数i - 男生人数j）+1小于n且新的差值（就是加了一个女生后）<=k，就可以转移；

#include<cstdio>
#include<iostream>
#define R register int
using namespace std;
const int N=;
int f[N<<][N][][],n,m,ans,cnt,k;
const int mod=;
signed main() {
    scanf("%d%d%d",&n,&m,&k);
    f[][][][]=;
    for(R i=;i<=n+m;++i) for(R j=;j<=n;++j)
        for(R o=;o<=k;++o) for(R p=;p<=k;++p) if(f[i][j][o][p]) {
            if(j+<=n&&o+<=k) (f[i+][j+][o+][max(p-,)]+=f[i][j][o][p])%=mod;
            if(i+-j<=m&&p+<=k) (f[i+][j][max(o-,)][p+]+=f[i][j][o][p])%=mod;
        }
    for(R i=;i<=k;++i) for(R j=;j<=k;++j) (ans+=f[n+m][n][i][j])%=mod;
    printf("%d\n",ans);
}

2019.3.14