题目描述
众所周知,“八皇后” 问题是求解在国际象棋棋盘上摆放 8个皇后,使得两两之间互不攻击的方案数。
已经学习了很多算法的小蓝觉得 “八皇后” 问题太简单了,意犹未尽。
作为一个国际象棋迷,他想研究在 N × 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 2 1
输出样例1
2
输入样例2
4 4 3
输出样例2
276
输入样例3
3 20 12
输出样例3
914051446
数据范围
对于 5 55% 的评测用例,K = 1 K=1K=1;
对于另外 10 1010% 的评测用例,K = 2 K=2K=2;
对于另外 10 1010% 的评测用例,N = 1 N=1N=1;
对于另外 20 2020% 的评测用例,N , M ≤ 6 , K ≤ 5 N,M≤6,K≤5N,M≤6,K≤5;
对于另外 25 2525% 的评测用例,N ≤ 3 , M ≤ 20 , K ≤ 12 N≤3,M≤20,K≤12N≤3,M≤20,K≤12;
对于所有评测用例,1 ≤ N ≤ 6 , 1 ≤ M ≤ 100 , 1 ≤ K ≤ 20 1≤N≤6,1≤M≤100,1≤K≤201≤N≤6,1≤M≤100,1≤K≤20。
思路:
状态压缩DP,dp[i][j][k][t]表示第i行共有j个马并且i-1行的状态为k,i-行的状态为t时摆放方式。
#include <bits/stdc++.h>
using namespace std;
const int M = 1 << 6, MOD = 1e9 + 7;
int n, m, K;
int dp[110][21][M][M];
int count(int x)
{
int res = 0;
while (x)
{
res++;
x &= (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 b = 0; b < 1 << n; b ++)
{
for (int a = 0; a < 1 << n; a ++)
{
if (b & (a << 2) || b & (a >> 2))
{
continue;
}
for (int c = 0; c < 1 << n; c ++)
{
if (c & (a << 1) || c & (a >> 1))
{
continue;
}
if (c & (b << 2) || c & (b >> 2))
{
continue;
}
int temp = count(a);
if (k >= temp)
{
dp[i][k][b][a] = (dp[i][k][b][a] + dp[i - 1][k - temp][c][b]) % MOD;
}
}
}
}
}
}
int ans = 0;
for (int i = 0; i < 1 << n; i ++)
{
for (int j = 0; j < 1 << n; j ++)
{
ans = (ans + dp[m][K][i][j]) % MOD;
}
}
cout << ans << endl;
return 0;
}