蓝桥杯算法训练 N皇后问题

问题描述

　　在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。你的任务是，对于给定的N，求出有多少种合法的放置方法。

输入格式

　　输入中有一个正整数N≤10，表示棋盘和皇后的数量

输出格式

　　为一个正整数，表示对应输入行的皇后的不同放置数量。

样例输入

5

样例输出

10

数据规模和约定

　　N≤10

思路：

深搜+回溯，关于对角线，只需要建立两个数组d1，d2，d1存储x+y，d2存储y-x+n，保证所有皇后的坐标x+y不相等，y-x+n不相等即可确保对角线上不存在皇后。

代码：

#include<bits/stdc++.h>
using namespace std;
int n,ans=0;
int l[15],d1[22],d2[22];
void dfs(int h)//h代表行
{
	if(h>n)//当h大于n就返回
	{
		ans++;
		return;
	}
	int i;
	for(i=1;i<=n;i++)//遍历列
	{
		if(l[i]||d1[h+i]||d2[i-h+n])	continue;//如果当前列已有皇后或者皇后的坐标x+y，y-x+n已有，则不继续搜索
		l[i]=1;
		d1[h+i]=1;
		d2[i-h+n]=1;
		dfs(h+1);
		l[i]=0;
		d1[h+i]=0;
		d2[i-h+n]=0;//回溯
	}
}
int main()
{
	cin>>n;
	dfs(1);
	cout<<ans;
	return 0;
}