传送门;牛客
题目描述:
在N行M列的棋盘上,放若干个炮可以是0个,使得没有任何一个炮可以攻击另一个炮。 请问有多少种放置方法,中国
像棋中炮的行走方式大家应该很清楚吧.一个炮要能攻击另一个炮他们必须要处于同一行或者一列且他们之间有且仅有
一个棋子.
输入:
1 3
输出:
7
靠,为什么这道题能出现在状压dp的题单里啊,明明这道题状压只能的部分分…
这道题的 m , n m,n m,n的范围达到了100,所以显然我们是不能直接使用状压dp来解决这道题的(害我想了很久如何构造状态WAW)
主要思路:
- 这道题就是一道状态方程比较难想的dp题.首先我们观察一下题目,我们会发现这道题抽象出来就是每一行,每一列不能同时存在两个炮.那么想一下,对于我们当前枚举到的第 i i i行,我们如果考虑第 i i i行放炮的位置的话,就需要知道前 i − 1 i-1 i−1行的棋子的状态.那么我们将思路转向如何记录前 i − 1 i-1 i−1行的炮的状态呢
- 所以我们就可以使用 d p [ i ] [ j ] [ k ] dp[i][j][k] dp[i][j][k]来记录前 i i i行有 j j j列是有一个炮存在的,有 k k k列是有两个炮存在的,那么如果这样想的话,我们的状态转移方程也就不难想出了
对于第 i i i行来说,我们有这几种可能性:
第 i i i行不准备放炮:
对于这一种情况来说,我们的转移十分的简单,我们的炮存在的情况并没有发生变化,所以直接转移即可,也就是 d p [ i + 1 ] [ j ] [ k ] = ( d p [ i + 1 ] [ j ] [ k ] + d p [ i ] [ j ] [ k ] ) % m o d dp[i+1][j][k]=(dp[i+1][j][k]+dp[i][j][k])\%mod dp[i+1][j][k]=(dp[i+1][j][k]+dp[i][j][k])%mod
第 i i i行准备放一个炮
对于这一种情况来说有两种可能性,一种是我们的这一个炮放在本来没有炮的列中,也就是:
d
p
[
i
+
1
]
[
j
+
1
]
[
k
]
=
(
d
p
[
i
+
1
]
[
j
+
1
]
[
k
]
+
d
p
[
i
]
[
j
]
[
k
]
∗
(
m
−
j
−
k
)
)
%
m
o
d
;
dp[i+1][j+1][k]=(dp[i+1][j+1][k]+dp[i][j][k]*(m-j-k))\%mod;
dp[i+1][j+1][k]=(dp[i+1][j+1][k]+dp[i][j][k]∗(m−j−k))%mod;
另一种是这一个炮放在本来就有一个炮的列中,也就是:
d
p
[
i
+
1
]
[
j
−
1
]
[
k
+
1
]
=
(
d
p
[
i
+
1
]
[
j
−
1
]
[
k
+
1
]
+
d
p
[
i
]
[
j
]
[
k
]
∗
(
j
)
)
%
m
o
d
;
dp[i+1][j-1][k+1]=(dp[i+1][j-1][k+1]+dp[i][j][k]*(j))\%mod;
dp[i+1][j−1][k+1]=(dp[i+1][j−1][k+1]+dp[i][j][k]∗(j))%mod;
第
i
i
i行准备放两个炮
这种情况有三种可能性,一种是两个炮都放在没有炮的列中,也就是(用到了组合数)
d
p
[
i
+
1
]
[
j
+
2
]
[
k
]
=
(
d
p
[
i
+
1
]
[
j
+
2
]
[
k
]
+
d
p
[
i
]
[
j
]
[
k
]
∗
C
(
m
−
j
−
k
)
)
%
m
o
d
;
dp[i+1][j+2][k]=(dp[i+1][j+2][k]+dp[i][j][k]*C(m-j-k))\%mod;
dp[i+1][j+2][k]=(dp[i+1][j+2][k]+dp[i][j][k]∗C(m−j−k))%mod;
或者一个炮放在没有炮的列中,一个炮放在有一个炮的列中,也就是:
d p [ i + 1 ] [ j − 2 ] [ k + 2 ] = ( d p [ i + 1 ] [ j − 2 ] [ k + 2 ] + d p [ i ] [ j ] [ k ] ∗ C ( j ) ) % m o d ; dp[i+1][j-2][k+2]=(dp[i+1][j-2][k+2]+dp[i][j][k]*C(j))\%mod; dp[i+1][j−2][k+2]=(dp[i+1][j−2][k+2]+dp[i][j][k]∗C(j))%mod;
或者两个炮都放在已经有一个炮的列中,也就是:
d p [ i + 1 ] [ j − 2 ] [ k + 2 ] = ( d p [ i + 1 ] [ j − 2 ] [ k + 2 ] + d p [ i ] [ j ] [ k ] ∗ C ( j ) ) % m o d ; dp[i+1][j-2][k+2]=(dp[i+1][j-2][k+2]+dp[i][j][k]*C(j))\%mod; dp[i+1][j−2][k+2]=(dp[i+1][j−2][k+2]+dp[i][j][k]∗C(j))%mod;
至此我们的情况就搞完了,对于最终答案我们只要累加一下即可
下面是具体的代码部分:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
#include <deque>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
ll x=0,w=1;char ch=getchar();
for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x*w;
}
#define int long long
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
int m,n;int dp[105][105][105];
const int mod=9999973;
int C(int N) {return N*(N-1)/2;}
signed main() {
n=read();m=read();
dp[0][0][0]=1;
for(int i=0;i<n;i++) {
for(int j=0;j<=m;j++) {
for(int k=0;k+j<=m;k++) {
if(dp[i][j][k]) {
//不放
dp[i+1][j][k]=(dp[i+1][j][k]+dp[i][j][k])%mod;
//放一个在没炮的列
if(m-j-k>=1) dp[i+1][j+1][k]=(dp[i+1][j+1][k]+dp[i][j][k]*(m-j-k))%mod;
//放一个在有一个炮的列
if(j>=1) dp[i+1][j-1][k+1]=(dp[i+1][j-1][k+1]+dp[i][j][k]*(j))%mod;
//放两个在没有炮的列
if(m-j-k>=2) dp[i+1][j+2][k]=(dp[i+1][j+2][k]+dp[i][j][k]*C(m-j-k))%mod;
//放两个一个放在有一个炮的列,一个放在没有炮的列
if(m-j-k>=1&&j>=1) dp[i+1][j][k+1]=(dp[i+1][j][k+1]+dp[i][j][k]*(j)*(m-j-k))%mod;
//放两个,两个都放在一个炮的列
if(j>=2) dp[i+1][j-2][k+2]=(dp[i+1][j-2][k+2]+dp[i][j][k]*C(j))%mod;
}
}
}
}
int ans=0;
for(int i=0;i<=m;i++) {
for(int j=0;i+j<=m;j++) {
ans=(ans+dp[n][i][j])%mod;
}
}
cout<<ans<<endl;
return 0;
}