题目描述
地球人都知道,在国际象棋中,后如同太阳,光芒四射,威风八面,它能控制横、坚、斜线位置。
看过清宫戏的中国人都知道,后宫乃步步惊心的险恶之地,各皇后都有自己的势力范围,但也总能找到相安无事的办法。
所有中国人都知道,皇权神圣,伴君如伴虎,触龙颜者死…
现在有一个 n × n 的皇宫,国王占据他所在位置及周围的共 9 个格子,
这些格子皇后不能使用(如果国王在王宫的边上,占用的格子可能不到 9 个),当然,皇后也不会攻击国王。
现在知道了国王的位置(x,y),国王位于第 x 行第 y 列,起始行和列均为 1。
请问,有多少种方案放置 n 个皇后,使她们不能互相攻击。
输入格式
三个整数 n,x,y,分别表示皇宫的规模及表示国王的位置
输出格式
一个整数,表示放置 n 个皇后的方案数
样例输入
8 2 2
样例输出
10
数据范围
n ≤ 12
题解
DFS:
只要先把国王的范围标记出来,就转变成了传统的八皇后问题。需要注意的是:由于需要记录对角线上的皇后,所以数组下标至少要开到 2n。
#include <iostream>
using namespace std;
const int N = 50;
int n, x, y, ans;
bool st[N][N], c[N], d[N], ud[N];
void dfs(int u)
{
if(u == n + 1)
{
ans ++;
return;
}
for (int i = 1; i <= n; i ++)
{
if(st[u][i] || c[i] || d[i - u + n] || ud[i + u]) continue;
c[i] = d[i - u + n] = ud[i + u] = true;
dfs(u + 1);
c[i] = d[i - u + n] = ud[i + u] = false;
}
}
int main()
{
cin >> n >> x >> y;
for (int i = x - 1; i <= x + 1; i ++)
for (int j = y - 1; j <= y + 1; j ++)
st[i][j] = true;
dfs(1);
cout << ans << endl;
return 0;
}