蓝桥杯试题 算法训练 王、后传说

蓝桥杯试题 算法训练 王、后传说

题目描述：

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　地球人都知道，在国际象棋中，后如同太阳，光芒四射，威风八面，它能控制横、坚、斜线位置。
　　看过清宫戏的中国人都知道，后宫乃步步惊心的险恶之地。各皇后都有自己的势力范围，但也总能找到相安无事的办法。
　　所有中国人都知道，皇权神圣，伴君如伴虎，触龙颜者死…
　　现在有一个n*n的皇宫，国王占据他所在位置及周围的共9个格子，这些格子皇后不能使用（如果国王在王宫的边上，占用的格子可能不到9个）。当然，皇后也不会攻击国王。
　　现在知道了国王的位置（x,y）（国王位于第x行第y列，x,y的起始行和列为1），请问，有多少种方案放置n个皇后，使她们不能互相攻击。
输入格式
　　一行，三个整数，皇宫的规模及表示国王的位置
输出格式
　　一个整数，表示放置n个皇后的方案数
样例输入
8 2 2
样例输出
10
数据规模和约定
　　n<=12
　　
注意n个皇后摆放需要满足的条件：

（1）每2个皇后不能放在同一行，同一列，同一个斜线（两个点组成的线的斜率的绝对值为1）。
（2）皇帝的位置以及周围的位置（一共9个位置）不能摆放皇后。

解题思路：

由上面的摆放条件可以知道，每2个皇后不能在同一行，和同一列，那么我们可以把问题转化为这样：找出1~n的全排列作为皇后摆放的列数的序列（行数是固定从1—>n的，方便计算和递归），当然这些全排列序列需要满足每两个位置的（第几行，第几列）的斜率绝对值不能是1，且不能在皇帝所在的位置。

AC代码：

#include <stdlib.h>
#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <math.h>
using namespace std;
int n;//皇宫的规模 
int ans=0;//最后的答案，即放置n个皇后的方案数 
int a[15];//a[i]的值表示，在第step行的第 a[i]列摆放皇后 
int vis[15];//标记该位置是否访问过，0表示为没有被访问 
void f(int step,int x,int y)//step表示给第step行摆放皇后 
{
	int i,j,flag;
	if(step==n+1)//n个皇后都摆放完毕 
	{
		ans++;
	} 
	else
	{
		for(i=1;i<=n;i++)
		{
			if(step>=x-1&&step<=x+1&&i>=y-1&&i<=y+1)//皇帝的9个位置，是不能摆放皇后的 
				continue;
			if(!vis[i])//没有被访问过 
			{
				flag=0;//我们这个时候需要判断，在该位置（step，i）没有被访问的情况下，能不能摆放皇后 
				for(j=1;j<step;j++)
				{
					if(abs(step-j)==abs(i-a[j]))//在同一个斜线上，不能在该位置摆放皇后 
					{
						flag=1;//标记该位置虽然没有被访问过，但是依然不能放皇后 
						break;
					}
				}
				if(!flag)
				{
					a[step]=i;
					vis[i]=1;
					f(step+1,x,y);
					vis[i]=0;//表示已经摆放完第step+1~n个皇后了，这个时候需要取消标记 
				} 
			}
		}
	}
}
int main()
{	
	int x,y;//皇帝的坐标 
	cin>>n>>x>>y;
	f(1,x,y);
	cout<<ans<<endl;
	return 0;
}