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

BZOJ & Luogu


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

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

两种转移:

if(j+1<=n&&o+1<=k) (f[i+1][j+1][o+1][max(p-1,0)]+=f[i][j][o][p])%=mod;
if(i+1-j<=m&&p+1<=k) (f[i+1][j][max(o-1,0)][p+1]+=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=160;
int f[N<<1][N][21][21],n,m,ans,cnt,k;
const int mod=12345678;
signed main() {
    scanf("%d%d%d",&n,&m,&k);
    f[0][0][0][0]=1;
    for(R i=0;i<=n+m;++i) for(R j=0;j<=n;++j) 
        for(R o=0;o<=k;++o) for(R p=0;p<=k;++p) if(f[i][j][o][p]) {
            if(j+1<=n&&o+1<=k) (f[i+1][j+1][o+1][max(p-1,0)]+=f[i][j][o][p])%=mod;
            if(i+1-j<=m&&p+1<=k) (f[i+1][j][max(o-1,0)][p+1]+=f[i][j][o][p])%=mod;
        }
    for(R i=0;i<=k;++i) for(R j=0;j<=k;++j) (ans+=f[n+m][n][i][j])%=mod;
    printf("%d\n",ans);
}

2019.3.14

 

posted @ 2019-03-14 21:00  LuitaryiJack  阅读(235)  评论(0编辑  收藏  举报