P1219 八皇后问题

题面：

[USACO1.5]八皇后 Checker Challenge - 洛谷

思路：

开四个数组，分别记录行、列、正对角线（左上到右下）、反对角线（右上到左下）是否有皇后，如果有则继续进行深搜，没有则回溯。

正对角线的数学原理：

以4X4棋盘，点（3，2）为例。

 

显然，正对角线上的点纵坐标减横坐标的值相等，即y-x的值相等。

反对角线的数学原理：

以4X4棋盘，点（3，2）为例。

 

反对角线上的点，横坐标与纵坐标之和相等，即y+x的值相等。

代码：

#include<iostream>
using namespace std;
const int N=100;
int a[N],b[N],c[N],d[N];//行，列，正对角线，反对角线
int n;
int sum;

void print() {
	if(sum<3) {
		for(int k=1; k<=n; k++)cout<<a[k]<<" ";
		cout<<endl;
	}
	sum++;
};

void queen(int x) {
	if(x>n) {
		print();
		return;
	}
	else {
		for(int y=1; y<=n; y++) {
			if((!b[y])&&(!c[y-x+n])&&(!d[x+y])) {
				a[x]=y;
				b[y]=1;
				//正对角线是(y-x)相同
				c[y-x+n]=1;
				//反对角线是(x+y)相同
				d[x+y]=1;
				//下一个皇后
				queen(x+1);
				//回溯
				b[y]=0;
				c[y-x+n]=0;
				d[x+y]=0;
			}
		}
	}
};

int main() {
	cin>>n;
	queen(1);
	cout<<sum;
	return 0;
};