[Bzoj1037][ZJOI2008]生日聚会(DP)
Description
Solution
这题状态比较难想,
\(dp[i][j][g][h]\)表示强i个人有j个男生,在某个区间男生最多比女生多g人,女生最多比男生多h人的方案数,然后DP即可
最后答案是是所有\(dp[n+m][n][i][j](0\leq i,j\leq k)\)加起来
Code
#include <cstdio>
#include <algorithm>
using namespace std;
const int mo=12345678;
int n,m,k,dp[310][160][23][23],Ans;
int main(){
scanf("%d%d%d",&n,&m,&k);
dp[0][0][0][0]=1;
for(int i=0;i<=n+m-1;++i)
for(int j=0;j<=n;++j)
for(int g=0;g<=k;++g)
for(int h=0;h<=k;++h){
if(j+1<=n&&g+1<=k) (dp[i+1][j+1][g+1][max(h-1,0)]+=dp[i][j][g][h])%=mo;
if(i-j+1<=m&&h+1<=k) (dp[i+1][j][max(g-1,0)][h+1]+=dp[i][j][g][h])%=mo;
}
for(int i=0;i<=k;++i)for(int j=0;j<=k;++j)(Ans+=dp[n+m][n][i][j])%=mo;
printf("%d\n",Ans);
return 0;
}