bzoj1037: [ZJOI2008]生日聚会Party

题目
题解

用f[i][j][x][y]表示到第i个人,总共j个男生,以i结尾的任意长度中男生最多比女生多x,女生最多比男生多y的方案数。
状态转移如下:
如果x+1<=k,f[i+1,j+1,x+1,max(y-1,0)]+=f[i][j][x][y]
如果y+1<=k,f[i+1,j,max(x-1,0),y+1]+=f[i][j][x][y]
最终答案为∑(0≤i≤k)∑(0≤j≤k)f[n+m][n][i][j]。

#include<bits/stdc++.h>
using namespace std;
const int M=12345678;
int n,m,k,i,j,x,y,ans,f[302][152][22][22];
int main(){
    scanf("%d%d%d",&n,&m,&k);
    f[0][0][0][0]=1;
    for (i=0;i<n+m;i++)
        for (j=0;j<=n;j++)
            for (x=0;x<=k;x++)
                for (y=0;y<=k;y++){
                    if (x+1<=k && j+1<=n) (f[i+1][j+1][x+1][max(y-1,0)]+=f[i][j][x][y])%=M;
                    if (y+1<=k && i+1-j<=m) (f[i+1][j][max(x-1,0)][y+1]+=f[i][j][x][y])%=M;
                }
    for (i=0;i<=k;i++)
        for (j=0;j<=k;j++) (ans+=f[n+m][n][i][j])%=M;
    printf("%d",ans);
}

猜你喜欢

转载自blog.csdn.net/xumingyang0/article/details/80739145