[BZOJ1037]-DP-状态定义

说在前面

前一天自己独立想了很久，未果。选择向别人的题解低头=w=
然后…脑子不清醒的时候…果然还是不要写这种题

---

题目

BZOJ1037传送门

题意

有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() ;
}