n个男生,m个女生排成一排,要求任意子段 中男女生的个数差 <=K
方案数?
f[i][j][k][l] 目前i个男生,j个女生,个数差值为 k 和 l
转移很明显,
但最好用刷表法
#include <iostream>
#include <cstring>
#include <cmath>
using namespace std ;
#define int long long
#define mod 12345678
int n,m,p;
int f[154][154][24][24];
int max(int x,int y){
return x<y?y:x;
}
void solve(){
int i,j,k,l;
f[0][0][0][0]=1;
for(i=0;i<=n;i++)
for(j=0;j<=m;j++)
for(k=0;k<=p;k++)
for(l=0;l<=p;l++){
f[i][j][k][l]+=f[i-1][j][]
}
if(f[i][j][k][l]){
int t=f[i][j][k][l];
f[i+1][j][k+1][max(l-1,0)]+=t; f[i+1][j][k+1][max(l-1,0)]%=mod;
f[i][j+1][max(k-1,0)][l+1]+=t; f[i][j+1][max(k-1,0)][l+1]%=mod;
}
int ans=0;
for(i=0;i<=p;i++)
for(j=0;j<=p;j++){
ans+=f[n][m][i][j],ans%=mod;
}
cout<<ans;
}
signed main(){
int i,j;
cin>>n>>m>>p; solve();
}