洛谷 P2051 [AHOI2009] 中国象棋 题解

Announcement

• Programmed on 2024/3/1
• Written on 2024/3/1

题目来源

• 洛谷 P2051

Description

给定一个 $n\times m$ 的棋盘，需要放置若干棋子，使得每行、每列至多只有 $2$ 个棋子，求放置棋子的方案数。

• $1\le n,m\le 100$。

Solution

首先想到动态规划，考虑如何设计动态规划的状态。

由于对于放置棋子问题一般按照“行”来转移，因此对于行的限制很容易实现。

于是考虑存储有关列的限制的信息。

那么我们可以令 $f[i][j][k]$ 表示前 $i$ 行中，已经放置了 $j$ 列只有一个棋子的列和 $k$ 列有两个棋子的列（即有 $m-j-k$ 个空列）。

考虑如何进行转移。分类讨论：

1. 一个棋子都不放：

$$f[i][j][k]+=f[i-1][j][k]$$

2. 把一个放在空列：

$$f[i][j][k]+=f[i-1][j-1][k]\ast (m-(j-1)\ast k)$$

3. 把一个放在非空列：

$$f[i][j][k]+=f[i-1][j+1][k-1]\ast (j+1)$$

4. 把两个都放在不同空列：

$$f[i][j][k]+=f[i-1][j-2][k]\ast C_{m-(j-2)-k}^2$$

5. 把一个放在空列、一个放在非空列：

$$f[i][j][k]+=f[i-1][j][k-1]\ast (m-j-(k-1))\ast j$$

6. 把两个放在不同的非空列：

$$f[i][j][k]+=f[i][j+2][k-2]\ast C_{j+2}^2$$

初始化 $f[0][0][0]=1$，答案为 $\sum _{i=0}^m\sum _{j=0}^{m-i}f[n][i][j]$。

Conclusion

• 对于有限制的棋盘放置问题，一般考虑按照行来转移。
• 若无法状压且不关心顺序，可以用“符合条件的数量”来转移。

Code

#include <bits/stdc++.h>
using namespace std;
const int p=9999973;
int n,m;
long long f[105][105][105]; // f[i][j][k]-前i行有j列有1个棋子，k列2个棋子的方案数
int calc(int x){ // C_x^2
    return x*(x-1)/2%p;
}
int main(){
    scanf("%d%d",&n,&m);
    f[0][0][0]=1;
    for (int i=1;i<=n;i++)
        for (int j=0;j<=m;j++)
            for (int k=0;k<=m-j;k++){
                (f[i][j][k]+=f[i-1][j][k])%=p; // 不放
                if (j>=1) (f[i][j][k]+=f[i-1][j-1][k]*(m-(j-1)-k)%p)%=p; // 放一个在空列
                if (k>=1) (f[i][j][k]+=f[i-1][j+1][k-1]*(j+1)%p)%=p; // 放一个在非空列
                if (j>=2) (f[i][j][k]+=f[i-1][j-2][k]*calc(m-(j-2)-k)%p)%=p; // 放2个在不同空列
                if (k>=1) (f[i][j][k]+=f[i-1][j][k-1]*(m-j-(k-1))*j%p)%=p; // 放1个在空列、1个在非空列
                if (k>=2) (f[i][j][k]+=f[i-1][j+2][k-2]*calc(j+2)%p)%=p; // 放2个在2个非空列
            }
    int ans=0;
    for (int i=0;i<=m;i++) for (int j=0;j<=m-i;j++) (ans+=f[n][i][j])%=p;
    printf("%d\n",ans);
    return 0;
}