【问题描述】
众所周知,“八皇后” 问题是求解在国际象棋棋盘上摆放 8 个皇后,使得两
两之间互不攻击的方案数。已经学习了很多算法的小蓝觉得 “八皇后” 问题太简
单了,意犹未尽。作为一个国际象棋迷,他想研究在 N × M 的棋盘上,摆放 K
个马,使得两两之间互不攻击有多少种摆放方案。由于方案数可能很大,只需
计算答案除以 1000000007 (即 + 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 (即 109 + 7) 的余数。
【样例输入】
1 2 1
【样例输出】
2
【样例输入】
4 4 3
【样例输出】
276
【样例输入】
3 20 12
【样例输出】
914051446
【评测用例规模与约定】
对于 5% 的评测用例,K = 1;
对于另外 10% 的评测用例,K = 2;
对于另外 10% 的评测用例,N = 1;
对于另外 20% 的评测用例,N, M ≤ 6,K ≤ 5;
对于另外 25% 的评测用例,N ≤ 3,M ≤ 20,K ≤ 12;
对于所有评测用例,1 ≤ N ≤ 6,1 ≤ M ≤ 100,1 ≤ K ≤ 20。
分析:DFS+回溯
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
using PII = pair<int, int>;
const int mod = 1000000007;
const int N = 10, M = 110;
int n, m, k;
int dire[8][2] = { {1,2}, {1,-2}, {-1,2}, {-1,-2}, {2,1}, {2,-1}, {-2,1} ,{-2,-1} }; //转移位置
int st[N][M]; //st[i][j] = true 表示 不能放 马,否则可以放
int res = 0;
void dfs(int x, int y, int num) //像数独一样枚举即可 从左到右,从上到下
{
if (x == n && y == 0) //终点
return;
dfs(x + (y + 1) / m, (y + 1) % m, num); //不放 马
if (st[x][y] == 0) { //可以 放 马
if (num + 1 == k) { // 结果达成
res = (res + 1) % mod;
return;
}
vector<PII> path; //记录 这个放了 这个马 之后引起的变化
st[x][y] = true; //马的位置肯定是true
for (int i = 0; i < 8; ++i) {
int xi = dire[i][0] + x, yi = dire[i][1] + y;
if (xi < n && xi >= 0 && yi < m && yi >= 0 && st[xi][yi] == 0) {
path.push_back({ xi,yi });
st[xi][yi] = 1;
}
}
dfs(x + (y + 1) / m, (y + 1) % m, num + 1); //访问
st[x][y] = false; //回溯
for (auto &a : path) //回溯
st[a.first][a.second] = 0;
}
}
int main()
{
cin >> n >> m >> k;
dfs(0, 0, 0);
cout << res << '\n';
return 0;
}