[蓝桥杯 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 题)。
首先,这数据量肯定是状压dp。然后,我们如果要取第i行,则影响该行的只有第i-1行以及第i-2行的状态。但是显然2^200肯定过不了,但是可以看到行的数量相对较少所以可以根据数到第几列来算。就可以定义一个数组s[p][n][a][b],p是数到了第几列,n是当前一共放了几个马,a是p-2列的状态,b是p-1列的状态。转移方程就是 s[p][n][a][b]=s[p][n][a][b]+s[p-1][n-c中1的个数][b][c],c为可能的第p-3列的状态。
#include<bits/stdc++.h>
using namespace std;
const int MOD=1e9+7;
int n,m,K,ans,dp[110][22][70][70];
int count(int x){
int res=0;
while(x){
res+=(x&1);
x>>=1;
}
return res;
}
int main(){
cin>>n>>m>>K;
dp[0][0][0][0]=1;
for(int i=1;i<=m;i++)
for(int k=0;k<=K;k++)
for(int a=0;a<(1<<n);a++){
for(int b=0;b<(1<<n);b++){
if((a&(b<<2))||(a&(b>>2))) continue;
for(int c=0;c<(1<<n);c++){
if((a&(c<<1))||(a&(c>>1))) continue;
if((b&(c<<2))||(b&(c>>2))) continue;
int t=count(a);
if(k>=t)
dp[i][k][b][a]=(dp[i][k][b][a]+dp[i-1][k-t][c][b])%MOD;
}
}
}
for(int a=0;a<(1<<n);a++)
for(int b=0;b<(1<<n);b++)
ans=(ans+dp[m][K][b][a])%MOD;
cout<<ans<<endl;
return 0;
}