说在前面
前一天自己独立想了很久,未果。选择向别人的题解低头=w=
然后…脑子不清醒的时候…果然还是不要写这种题
题目
题意
有N个男孩子M个女孩子排成一排,要求任意一段区间内男孩子和女孩子个数之差不能大于K,询问符合条件的排队方案数。答案对12345678取模。
输入与输出
输入:三个整数N,M,K。N,M≤150,K≤20
输出:答案对12345678取模后的数
解法
定义DP数组
dp[i][j][k][x]表示在前i个人当中,选取了j个男孩子,并且在这前i个人的任意长度后缀中,男孩子最多比女孩子多了k个,女孩子最多比男孩子多了x个时的方案数。
使用刷表法(填表法转移不好写),当前在dp[i][j][k][x];
下一个如果选的是男孩子,就转移到dp[i+1][j+1][k+1][max(x-1,0)]
下一个如果选的是女孩子,就转移到dp[i+1][j][max(k-1,0)][j+1]
初值dp[0][0][0][0]=1。最后的答案就是对dp[N+M][N]求和
注意写的时候需要判断边界,比如说”当前男孩子已经选了N个了,就不能再选男孩子了”之类的,一定要判清楚。
自带大常数的代码
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;
int N , M , K , dp[305][155][21][21] , mmod = 12345678 , ans ;
void solve(){
for( register int i = 0 ; i < N + M ; i ++ ){
for( register int j = 0 ; j <= N && j <= i ; j ++ ){
for( register int k = 0 ; k <= K && k <= j ; k ++ ){
for( register int x = 0 ; x <= K && x <= i - j ; x ++ ){
if( dp[i][j][k][x] >= mmod ) dp[i][j][k][x] %= mmod ;
if( k != K ) dp[i+1][j+1][k+1][max(x-1,0)] += dp[i][j][k][x] ;
if( x != K ) dp[i+1][j ][max(k-1,0)][x+1] += dp[i][j][k][x] ;
}
}
}
}
for( int k = 0 ; k <= K ; k ++ )
for( int x = 0 ; x <= K ; x ++ )
ans = ( ans + dp[N+M][N][k][x] )%mmod ;
printf( "%d" , ans ) ;
}
int main(){
scanf( "%d%d%d" , &N , &M , &K ) ;
dp[0][0][0][0] = 1 ;
solve() ;
}