国际象棋_

[蓝桥杯 2021 省 AB2] 国际象棋

题目描述

众所周知， “八皇后” 问题是求解在国际象棋棋盘上摆放 $8$ 个皇后，使得两两之间互不攻击的方案数。已经学习了很多算法的小蓝觉得 “八皇后” 问题太简单了，意犹末尽。作为一个国际象棋迷，他想研究在 $N\times M$ 的棋盘上，摆放 $K$ 个马，使得两两之间互不攻击有多少种摆放方案。由于方案数可能很大，只需计算答案除以 $1000000007$ (即 $10^9+7)$ 的余数。

如下图所示，国际象棋中的马摆放在棋盘的方格内，走 “日” 字, 位于 $(x,y)$ 格的马（第 $x$ 行第 $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)$ 共 $8$ 个 格子。

输入格式

输入一行包含三个正整数 $N,M,K$, 分别表示棋盘的行数、列数和马的个数。

输出格式

输出一个整数，表示摆放的方案数除以 $1000000007($ 即 $10^9+7)$ 的余数。

样例 #1

样例输入 #1

1 2 1

样例输出 #1

2

样例 #2

样例输入 #2

4 4 3

样例输出 #2

276

样例 #3

样例输入 #3

3 20 12

样例输出 #3

914051446

提示

对于 $5\%$ 的评测用例, $K=1$;

对于另外 $10\%$ 的评测用例, $K=2$;

对于另外 $10\%$ 的评测用例, $N=1$;

对于另外 $20\%$ 的评测用例, $N,M\le 6,K\le 5$;

对于另外 $25\%$ 的评测用例, $N\le 3,M\le 20，K\le 12$;

对于所有评测用例, $1\le N\le 6,1\le M\le 100,1\le K\le 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;
}