洛谷P2051 中国象棋【dp】

题目https://www.luogu.org/problemnew/show/P2051ios

题意:n*m的格子里放炮,使他们不能互相攻击。spa

若是两个炮在同一行同一列而且中间还有一个棋子的话就能够攻击。问有多少种放炮的方案。code

思路:首先根据规则,能够推出任意的行和列中炮的个数都不能超过2个。blog

能够尝试用递推。$dp[i][j][k]$表示处理到第$i$行,只有一个炮的列有$j$个,有两个炮的列有$k$个时的方案数。get

当$i-1$行处理完了处理$i$的时候,能够放0,1,2个炮。string

因此$dp[i][j][k] = dp[i-1][j][k] + dp[i-1][j-1][k] * (m - j - k + 1) + dp[i - 1][j + 1][k - 1] * (j + 1) + dp[i - 1][j - 2][k] * (m - j - k + 2)(m - j - k + 1) + dp[i - 1][j + 2][k - 2] * (j + 2)(j + 1)/2 + dp[i- 1][j][k - 1] * j(m - j - k + 1)$【每次像这种到总是会搞乱究竟是谁推出谁的关系】it

注意中间过程可能会爆int,因此直接用LL作吧。io

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<map>
 4 #include<set>
 5 #include<cstring>
 6 #include<algorithm>
 7 #include<vector>
 8 #include<cmath> 
 9 #include<stack>
10 #include<queue>
11 #include<iostream>
12 
13 #define inf 0x3f3f3f3f
14 using namespace std;
15 typedef long long LL;
16 typedef pair<int, int> pr;
17 
18 int n, m;
19 const int maxn = 105;
20 const LL mod = 9999973;
21 LL dp[maxn][maxn][maxn];
22 
23 int main()
24 {
25     scanf("%d%d", &n, &m);
26     dp[0][0][0] = 1;
27     for(int i = 1; i <= n; i++){
28         for(int j = 0; j <= m; j++){
29             for(int k = 0; j + k <= m; k++){
30                 dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j][k]) % mod;
31                 if(j >= 1)dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j - 1][k] * (m - j - k + 1)) % mod;
32                 if(k >= 1 && j + 1 <= m)dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j + 1][k - 1] * (j + 1)) % mod;
33                 if(j >= 2)dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j - 2][k] * ((m - j - k + 2) * (m - j - k + 1) / 2)) % mod;
34                 if(k >= 2 && j + 2 <= m)dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j + 2][k - 2] * ((j + 2) * (j + 1) / 2)) % mod;
35                 if(k >= 1)dp[i][j][k] = (dp[i][j][k] + dp[i - 1][j][k - 1] * j * (m - j - k + 1)) % mod;
36             }
37         }
38     }
39     
40     LL ans = 0;
41     for(int j = 0; j <= m; j++){
42         for(int k = 0; j + k <= m; k++){
43             ans = (ans + dp[n][j][k]) % mod;
44         }
45     }
46     printf("%lld\n", ans);
47     return 0;
48 }