洛谷 [P2051] 中国象棋

DP

orz__stdcall

首先要想出来,每行最多只能放两个棋子,这是显然的

于是决策就是一行一行地处理

30分的做法就是裸的枚举,暴搜,枚举这一行放哪里,放几个

然后想到了压位dp,按3进制表示当前棋盘的状态,即某一列没有棋子,或者有一个,两个棋子,能过50分

接着可以发现,棋子的顺序是无所谓的,并不需要准确知道当前棋盘的状态

于是有了100分做法:dp[i][j][k]表示放了前i行,有j列是有1个棋子,有k列有两个棋子

然后枚举所有的转移即可

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MOD = 9999973;
long long  n, m, dp[105][105][105];
long long C(long long num) {
    return num * (num - 1) / 2;
}
int main() {
    cin >> n >> m;
    dp[0][0][0] = 1;
    for(int i = 1; i <= n; i++) {
        for(int j = 0; j <= m; j++) {
            for(int k = 0; k + j <= m; k++) {
                dp[i][j][k] += dp[i - 1][j][k] % MOD;
                if(j > 0) (dp[i][j][k] += dp[i - 1][j - 1][k] * (m - j - k + 1)) %= MOD;
                if(j < m && k > 0) (dp[i][j][k] += dp[i - 1][j + 1][k - 1] * (j + 1)) %= MOD;
                if(j > 1) (dp[i][j][k] += dp[i - 1][j - 2][k] * C(m - j - k  + 2)) %= MOD;
                if(k > 0) (dp[i][j][k] += dp[i - 1][j][k - 1] * (m - j - k + 1) * j) %= MOD;
                if(j < m - 1 && k > 1) (dp[i][j][k] += dp[i - 1][j + 2][k - 2] * C(j + 2)) %= MOD;
            }
        }
    }
    long long ans = 0ll;
    for(int i = 0; i <= m; i++) {
        for(int j = 0; j + i <= m; j++) {
            (ans += dp[n][i][j]) %= MOD;
        }
    }
    cout << ans << endl;
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/Mr-WolframsMgcBox/p/9073685.html