題解 BZOJ P1037 & Luogu P2592 [ZJOI2008]生日聚會
阿新 • • 發佈:2019-03-14
pre cor names def ostream div tar .com 如果
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
題解 BZOJ P1037 & Luogu P2592 [ZJOI2008]生日聚會