Bzoj1037ZJOI2008生日聚会（DP）

Posted 2020-10-29 void_f

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;
}