王、后传说（升级八皇后问题）

王、后传说 (20分)

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

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

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

现在有一个n*n的皇宫，国王占据他所在位置及周围的共9个格子，这些格子皇后不能使用（如果国王在王宫的边上，占用的格子可能不到9个）。当然，皇后也不会攻击国王。

现在知道了国王的位置（x,y）（国王位于第x行第y列，x,y的起始行和列为1），

请问，有多少种方案放置n个皇后，使她们不能互相攻击。

输入格式:
一行，三个整数，皇宫的规模及表示国王的位置

输出格式:
一个整数，表示放置n个皇后的方案数

输入样例:
在这里给出一组输入。例如：

8 2 2

输出样例:
在这里给出相应的输出。例如：

10

1. 设置一个数组用来存放位置 ，给所有的位置赋值为0

int cnt=0;//记录方案数
int map[14][14]={0};//位置

2. 将国王所在位置及他周围的9个位置 赋值为1，这些位置是王后不能占领的
   

void  emp(int x,int y){
	for(int i=x-1;i<=x+1;i++){
		for(int j=y-1;j<=y+1;j++){
			map[i][j]=1;
		}
	}
}

3. 判断王后能不能占领一个位置，每一行、每一列、每一条斜线都只能有一个王后，所以王后占领的这个位置要满足：
   1）不能在已有王后的斜左下方
   2）不能在已有王后的右斜下方
   3）不能在已有王后的正下方
   

//判断王后的斜左上角、斜右上角、正上方是否有已经存在的王后
//参数：y：行 x：列 n：规模 
int panduan(int y,int x,int n){
	//如果该位置是国王所在位置 返回0 
	if(map[y][x]==1){
		return 0;
	} 
	int x2=x-1;
	int y2=y-1;
	//判断斜左上角
	while(x2>=1&&y2>=1){
		//如果map[x][y]=2，表示有其他王后存在，返回0 
		if(map[y2][x2]==2){
			return 0;
		}
		//如果没有，就继续往上面看有没有其他的王后
		x2--;
		y2--; 
	} 
	
	//判断斜右上角
	x2=x+1;
	y2=y-1;
	while(x2<=n&&y2>=1){
		if(map[y2][x2]==2)
		return 0;
		x2++;
		y2--;
	} 
	
	//判断正上方
	 y2=y-1;
	 while(y2>=1){
	 	if(map[y2][x]==2)//注意是x不是x2 
	 	return 0;
	 	y2--;
	 }
	 return 1;
} 

4. 遍历位置寻找王后的位置，将王后的位置赋值为2

//y:当前所在行数 ,n：规模 
void dfs(int y,int n){
	//如果行数等于规模数+1  ，cnt++
	if(y==n+1){
		cnt++;
		
	} 
	else{
		for(int i=1;i<=n;i++){
		//如果第y行，第i列可以放置一个王后，则继续向下找下一个可以放置王后的位置 
		if(panduan(y,i,n)){
			map[y][i]=2;
			dfs(y+1,n) ;
			//当找完一次方案之后，要将之前设置为2的位置全部重新设置成0，寻找下一个方案
			map[y][i]=0; 
		}
	}
	}
} 

完整代码

#include<iostream>
#include<algorithm>
using namespace std;
int cnt=0;
int map[14][14]={0};
//国王所在的位置设置为1 
void  emp(int x,int y){
	for(int i=x-1;i<=x+1;i++){
		for(int j=y-1;j<=y+1;j++){
			map[i][j]=1;
		}
	}
}
//判断王后的斜左上角、斜右上角、正上方是否有已经存在的王后
//参数：y：行 x：列 n：规模 
int panduan(int y,int x,int n){
	//如果该位置是国王所在位置 返回0 
	if(map[y][x]==1){
		return 0;
	} 
	int x2=x-1;
	int y2=y-1;
	//判断斜左上角
	while(x2>=1&&y2>=1){
		//如果map[x][y]=2，表示有其他王后存在，返回0 
		if(map[y2][x2]==2){
			return 0;
		}
		//如果没有，就继续往上面看有没有其他的王后
		x2--;
		y2--; 
	} 
	
	//判断斜右上角
	x2=x+1;
	y2=y-1;
	while(x2<=n&&y2>=1){
		if(map[y2][x2]==2)
		return 0;
		x2++;
		y2--;
	} 
	
	//判断正上方
	 y2=y-1;
	 while(y2>=1){
	 	if(map[y2][x]==2)//注意是x不是x2 
	 	return 0;
	 	y2--;
	 }
	 return 1;
} 
//y:当前所在行数 ,n：规模 
void dfs(int y,int n){
	//如果行数等于规模数+1  ，cnt++
	if(y==n+1){
		cnt++;
		
	} 
	else{
		for(int i=1;i<=n;i++){
		//如果第y行，第i列可以放置一个王后，则继续向下找下一个可以放置王后的位置 
		if(panduan(y,i,n)){
			map[y][i]=2;
			dfs(y+1,n) ;
			//当找完一次方案之后，要将之前设置为2的位置全部重新设置成0，寻找下一个方案
			map[y][i]=0; 
		}
	}
	}
} 

int main(){
	int a,b,n;
	cin>>n>>a>>b;
	emp(a,b);
	//每一行、每一列、每一斜线，只能有一个王后 
	//找每一列中的王后位置 
	for(int i=1;i<=n;i++){
		if(panduan(1,i,n)){
			map[1][i]=2;
			dfs(2,n);
			map[1][i]=0;
		}
	} 
	cout<<cnt;
}