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