中国象棋 \operatorname{中国象棋} 中国象棋
题目链接: luogu P2051 \operatorname{luogu\ P2051} luogu P2051 / SSL比赛 1465 \operatorname{SSL比赛\ 1465} SSL比赛 1465
题目
这次小可可想解决的难题和中国象棋有关。在一个 N N N 行 M M M 列的棋盘上,让你放若干个炮(可以是 0 0 0 个),使得没有任何一个炮可以攻击另一个炮,请问有多少种放置方法。大家肯定很清楚,在中国象棋中炮的行走方式是:一个炮能攻击到另一个炮,当且仅当它们在同一行或同一列中,且它们之间恰好有一个棋中。你也来和小可可一起锻炼一下思维吧!
输入
一行包含两个整数 N , M N, M N,M ,之间由一个空格隔开。
输出
总共的方案数。由于该值可能很大,只需给出方案数模 9999973 9999973 9999973 的结果。
样例输入
1 3
样例输出
7
样例解释
除了在 3 3 3 个格子里都塞满炮以外,其他方案都是可行的。所以一共有 2 3 − 1 = 7 2^3-1=7 23−1=7 种方案。
数据范围
30
%
30\%
30% 的数据中
N
N
N 和
M
M
M 均不超过
6
6
6
50
%
50\%
50% 的数据中
N
N
N 和
M
M
M 至少有一个数不超过
8
8
8
100
%
100\%
100% 的数据中
N
N
N 和
M
M
M 不超过
100
100
100
思路
这道题是一道状压 dp 。
我们设
f
[
i
]
[
j
]
[
k
]
f[i][j][k]
f[i][j][k] 为前
i
i
i 行,有
i
i
i 列有
1
1
1 个炮,有
k
k
k 列有
2
2
2 个炮。
那我们就枚举这三个,先保证
j
+
k
<
=
n
j + k <= n
j+k<=n ,这个不多说了吧。
接着就是转移方程了:
- 不放,就从 f [ i − 1 ] [ j ] [ k ] f[i - 1][j][k] f[i−1][j][k] 转移过来,方案就一种
- 在原来只有一个的地方放一个,就从 f [ i − 1 ] [ j + 1 ] [ k − 1 ] f[i - 1][j + 1][k - 1] f[i−1][j+1][k−1] 转移过来,方案有 ( j + 1 ) (j + 1) (j+1) 种
- 在原来没有的地方放一个,就从 f [ i − 1 ] [ j − 1 ] [ k ] f[i - 1][j - 1][k] f[i−1][j−1][k] 转移过来,方案有 m − ( j − 1 ) − k m - (j - 1) - k m−(j−1)−k 种
- 在原来只有一个的地方选两个位置各放一个,就从 f [ i − 1 ] [ j + 2 ] [ k − 2 ] f[i - 1][j + 2][k - 2] f[i−1][j+2][k−2] 转移过来,方案有 C ( j + 2 , 2 ) C(j+2,2) C(j+2,2) 种
- 在原来没有的地方选两个位置各放一个,就从 f [ i − 1 ] [ j − 2 ] [ k ] f[i - 1][j - 2][k] f[i−1][j−2][k] 转移过来,方案有 C ( m − ( j − 2 ) − k , 2 ) C(m - (j - 2) - k, 2) C(m−(j−2)−k,2) 种
- 在原来有一个和原来没有的地方各放一个,就从 f [ i − 1 ] [ j ] [ k − 1 ] f[i - 1][j][k - 1] f[i−1][j][k−1] 转移过来,方案有 j ∗ ( m − ( j − 1 ) − k ) j * (m - (j - 1) - k) j∗(m−(j−1)−k) 种。
我们只要把上面的加起来,就得到了
f
[
i
]
[
j
]
[
k
]
f[i][j][k]
f[i][j][k] 。
(当然,有一些超出边界的不要算进去)
要注意的就是要记得取模,而且求组合数我们可以直接求,因为都是 C ( x , 2 ) C(x, 2) C(x,2) ,可以等于 x ∗ ( x − 1 ) / 2 x * (x - 1) / 2 x∗(x−1)/2。
输出之前先统计 f [ n ] f[n] f[n] 中所有数(即枚举 j , k j,k j,k ),把这些数加起来,就是答案。
就没了。
代码
#include<cstdio>
#include<algorithm>
#define mo 9999973
#define ll long long
using namespace std;
ll n, m, a[101], f[101][101][101], ans;
//j为没有,k为有一个
int C(int now) {//求出C(now, 2)的值
return (now * (now - 1) / 2) % mo;
}
int main() {
scanf("%d %d", &n, &m);//读入
f[0][0][0] = 1;//初始化
for (ll i = 1; i <= n; i++)
for (ll j = 0; j <= m; j++)
for (ll k = 0; k <= m - j; k++) {
f[i][j][k] = f[i - 1][j][k];//不放
if (k >= 1) f[i][j][k] = (f[i][j][k] + (f[i - 1][j + 1][k - 1] * (j + 1)) % mo) % mo;//在原来有一个的地方放一个
if (j >= 1) f[i][j][k] = (f[i][j][k] + (f[i - 1][j - 1][k] * (m - (j - 1) - k)) % mo) % mo;//在原来没有的地方放一个
if (k > 1) f[i][j][k] = (f[i][j][k] + (f[i - 1][j + 2][k - 2] * C(j + 2)) % mo) % mo;//在原来有一个的两个不同地方放两个
if (j > 1) f[i][j][k] = (f[i][j][k] + (f[i - 1][j - 2][k] * C(m - (j - 2) - k)) % mo) % mo;//在原来没有的两个不同地方放两个
if (k >= 1 && j >= 1) f[i][j][k] = (f[i][j][k] + (f[i - 1][j][k - 1] * (j * (m - (j - 1) - k))) % mo) % mo;//在原来有一个和原来没有的地方各放一个
}
for (ll j = 0; j <= m; j++)
for (ll k = 0; k <= m - j; k++)
ans = (ans + f[n][j][k]) % mo;//统计所有的f[n]
printf("%lld", (ans + mo) % mo);//输出
return 0;
}