【AHOI2009】中国象棋

Description

给定一个棋盘， 在棋盘上放入m个炮，使得炮两两之间不能攻击，求出方案数。

Solution

考虑设计一个dp，定义f[i][j][k]表示前i行、有j列放了一个炮、有k列放了两个炮的方案数。

有一个显然的结论：如果方案合法，那么任何一行（列）的炮的数量最多为2。

然后我们思考转移，假设当前状态为(i, j, k)，我们有如下决策：

当前这一行可以一个炮也不放

如果一个炮都没放的列数量超过0，那么我们可以在这个位置上放一个炮

如果放一个炮的列的数量超过0，那么我们可以在这个位置上放一个炮

如果一个炮都没放的列数量超过1，那么我们可以在这个位置上放两个炮，方案数我们需要求一下组合数

如果放一个炮的列的数量超过0，且不放炮的列的数量超过0，那么我们可以在这两个位置上各放一个炮

如果放一个炮的列的数量超过1，那么我们可以在这里放两个炮，方案数我们需要求一下组合数

综上，我们讨论出了状态转移，时间复杂度为O(nm^2);

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, m;
ll f[110][110][110];
const int mod = 9999973;
int calc(int x) {
    return x * (x - 1) / 2;
}
int main() {
    scanf("%d%d", &n, &m);
    f[0][0][0] = 1;
    for (register int i = 0; i < n; ++i) {
        for (register int j = 0; j <= m; ++j) {
            for (register int k = 0; k <= m; ++k) {
                f[i + 1][j][k] = (f[i + 1][j][k] + f[i][j][k]) % mod;
                if (m - j - k > 0) f[i + 1][j + 1][k] = (f[i + 1][j + 1][k] + f[i][j][k] * (m - j - k)) % mod;
                if (j > 0) f[i + 1][j - 1][k + 1] = (f[i + 1][j - 1][k + 1] + f[i][j][k] * j) % mod;
                if (j > 1) f[i + 1][j - 2][k + 2] = (f[i + 1][j - 2][k + 2] + f[i][j][k] * calc(j)) % mod;
                if (j > 0 && m - j - k > 0) f[i + 1][j][k + 1] = (f[i + 1][j][k + 1] + f[i][j][k] * j * (m - j - k)) % mod;
                if (m - j - k > 1) f[i + 1][j + 2][k] = (f[i + 2][j + 2][k] + f[i][j][k] * calc(m - j - k)) % mod;
            }
        }
    }
    ll ans = 0;
    for (register int i = 0; i <= m; ++i)
        for (register int j = 0; j + i <= m; ++j)
            ans = (ans + f[n][i][j]) % mod;
    printf("%lld
", ans % mod);
    return 0;
}