特定条件的八皇后问题

7-4 特定条件的八皇后问题 (25分)
在国际象棋中，皇后是最厉害的棋子，可以横走、直走，还可以斜走。棋手马克斯·贝瑟尔 1848 年提出著名的八皇后问题：即在 8 × 8 的棋盘上摆放八个皇后，使其不能互相攻击 —— 即任意两个皇后都不能处于同一行、同一列或同一条斜线上。 要求第一行的皇后放在指定列，问有多少种摆法？

输入格式:
第一行的皇后放的列

输出格式:
解的个数

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

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

4

每一个棋子放下之后他的那一列（竖）和左斜方向（用撇表示），右斜方向（用捺表示）都不能再放其他棋子。而列只用将那一列的编号存入set中就能找到。根据简单的实验可以得到，同一撇上的所有数据横坐标+纵坐标的值是一样的，同理，每一捺上的横坐标-纵坐标的值也是一样的。根据这个原理可以将撇和捺保存入set中接下来就是简单的深度优先就能得到结果。八皇后问题也能使用这个方法解决。

#include<bits/stdc++.h>
using namespace std;
int a[10][10];
set<int> pie,na,shu;//pie(i+n),na(i-n)shu(n)
int all;
void dfs(int n){
	if(n==8){
		for(int i=1;i<9;i++){
			if(shu.find(i)==shu.end()&&pie.find(i+n)==pie.end()&&na.find(n-i)==na.end()){
				all++;
			}
		}
		return;
	}
	for(int i=1;i<9;i++){
		if(shu.find(i)==shu.end()&&pie.find(i+n)==pie.end()&&na.find(n-i)==na.end()){
			shu.insert(i);
			pie.insert(n+i);
			na.insert(n-i);
			dfs(n+1);
			shu.erase(i);
			pie.erase(n+i);
			na.erase(n-i);
		}
	}
	
}

int main(){
	for(int i=0;i<10;i++){
		for(int j=0;j<10;j++){
			a[i][j]=0;
		}
	}
	int n;
	cin>>n;
	a[1][n]=1;
	pie.insert(1+n);
	na.insert(1-n);
	shu.insert(n);
	dfs(2);
	cout<<all;
}