[蓝桥杯 2021 省 AB2] 国际象棋
题目描述
众所周知, “八皇后” 问题是求解在国际象棋棋盘上摆放 8 8 8 个皇后,使得两两之间互不攻击的方案数。已经学习了很多算法的小蓝觉得 “八皇后” 问题太简单了,意犹末尽。作为一个国际象棋迷,他想研究在 N × M N \times M N×M 的棋盘上,摆放 K K K 个马,使得两两之间互不攻击有多少种摆放方案。由于方案数可能很大,只需计算答案除以 1000000007 1000000007 1000000007 (即 1 0 9 + 7 ) \left.10^{9}+7\right) 109+7) 的余数。
如下图所示,国际象棋中的马摆放在棋盘的方格内,走 “日” 字, 位于 ( x , y ) (x, y) (x,y) 格的马(第 x x x 行第 y y y 列)可以攻击 ( x + 1 , y + 2 ) , ( x + 1 , y − 2 ) , ( x − 1 , y + 2 ) , ( x − 1 , y − 2 ) , ( x + 2 , y + 1 ) , ( x + 2 , y − 1 ) , ( x − 2 , y + 1 ) , ( x − 2 , y − 1 ) (x+1, y+2),(x+1, y-2),(x-1, y+2),(x-1, y-2),(x+2, y+1),(x+2, y-1),(x-2, y+1),(x-2, y-1) (x+1,y+2),(x+1,y−2),(x−1,y+2),(x−1,y−2),(x+2,y+1),(x+2,y−1),(x−2,y+1),(x−2,y−1) 共 8 8 8 个 格子。
输入格式
输入一行包含三个正整数 N , M , K N, M, K N,M,K, 分别表示棋盘的行数、列数和马的个数。
输出格式
输出一个整数,表示摆放的方案数除以 1000000007 ( 1000000007\left(\right. 1000000007( 即 1 0 9 + 7 ) \left.10^{9}+7\right) 109+7) 的余数。
样例 #1
样例输入 #1
1 2 1
样例输出 #1
2
样例 #2
样例输入 #2
4 4 3
样例输出 #2
276
样例 #3
样例输入 #3
3 20 12
样例输出 #3
914051446
提示
对于 5 % 5 \% 5% 的评测用例, K = 1 K=1 K=1;
对于另外 10 % 10 \% 10% 的评测用例, K = 2 K=2 K=2;
对于另外 10 % 10 \% 10% 的评测用例, N = 1 N=1 N=1;
对于另外 20 % 20 \% 20% 的评测用例, N , M ≤ 6 , K ≤ 5 N, M \leq 6, K \leq 5 N,M≤6,K≤5;
对于另外 25 % 25 \% 25% 的评测用例, N ≤ 3 , M ≤ 20 , K ≤ 12 N \leq 3, M \leq 20 , K \leq 12 N≤3,M≤20,K≤12;
对于所有评测用例, 1 ≤ N ≤ 6 , 1 ≤ M ≤ 100 , 1 ≤ K ≤ 20 1 \leq N \leq 6,1 \leq M \leq 100,1 \leq K \leq 20 1≤N≤6,1≤M≤100,1≤K≤20。
蓝桥杯 2021 第二轮省赛 A 组 I 题(B 组 J 题)。
思路
本道题第一眼看有点类似与小国王和炮兵阵地的题目,再看数据范围,都比较小,ok,我们就用状态压缩dp;
- 首先,我们分析状态表示,我们可以用
f[i][a][b][j]
表示第i行,第i行的状态a,第i-1行状态b,放了j匹马,然后我们可以画出以下图:
- 其次,状态转移方程就很好写了
-
⇒
\Rightarrow
⇒ 相邻两列的左右一位的位置上面不能同时有1
⇒ \Rightarrow ⇒ 间隔一列的左右两位的位置上面不能同时有1
a:i - 2列, b:i - 1列, c:i列
(a, b), (b, c)是相邻的一列, (a, c)中间间隔一列
t: 表示c中的二进制1的个数
然后就可以进行状态转移了,状态转移方程为:
f [ i , b , c , k ] + = f [ i − 1 , a , b , k − t ] f[i,b,c,k] += f[i - 1, a, b, k - t] f[i,b,c,k]+=f[i−1,a,b,k−t]
代码
#include<iostream>
#include<algorithm>
using namespace std;
//这道题跟炮兵阵地有点点区别,但区别不是太明显。
//因为炮兵阵地是得看前i-3行,而本道题只需看i-2行即可,因此状态就只看两层即可
const int N = 110,M = 1<<6,K = 22,mod=1e9+7;
int f[N][M][M][K];
int n,m,k;
int get(int x){
//算某个状态1的个数
int res=0;
while(x){
res++;
x-=x&-x;
}
return res;
}
int main(){
cin>>n>>m>>k;//n行数、m列数
f[0][0][0][0]=1;
//是一样的 一个棋盘N行M列 你转90°就是M行N列了 因为枚举列的状态就要超时了 N是6很小 所以枚举行
for(int i=1;i<=m;i++){
for(int a=0;a<(1<<n);a++){
for(int b=0;b<(1<<n);b++){
if(b&(a<<2)||a&(b<<2))continue;
for(int c=0;c<(1<<n);c++){
if(c&(b<<1)||b&(c<<1))continue;
if(c&(a<<2)||a&(c<<2))continue;
int t=get(b);
for(int j=t;j<=k;j++){
//其实后面加的那个就是能更新a,b状态的状态
f[i][a][b][j]=(f[i][a][b][j]+f[i-1][c][a][j-t])%mod;
}
}
}
}
}
int res=0;
for(int a=0;a<(1<<n);a++){
for(int b=0;b<(1<<n);b++){
res=(res+f[m][a][b][k])%mod;
}
}
cout<<res;
return 0;
}