算法训练(VIP)——王、后传说

题目描述
地球人都知道,在国际象棋中,后如同太阳,光芒四射,威风八面,它能控制横、坚、斜线位置。

看过清宫戏的中国人都知道,后宫乃步步惊心的险恶之地,各皇后都有自己的势力范围,但也总能找到相安无事的办法。

所有中国人都知道,皇权神圣,伴君如伴虎,触龙颜者死…

现在有一个 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;		
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值