n皇后问题

问题描述

在n×n的棋盘上放置彼此不受攻击的n个皇后。按照国际象棋的规则，皇后可以攻击与之处于同一行或同一列或同一斜线上的旗子。于是n后问题等价于在棋盘上放置n个皇后，任何2个皇后不在同一行或同一列或同一斜线上。n皇后问题有解的前提是仅当 皇后个数 = 1 或 n ≥ 4 。

算法设计

我们用n元数组x[1:n]表示n后问题的解，x[i]表示皇后i放置在第i行的第x[i]列，由于不允许将2个皇后放在同一列，所以向量中的x[i]互不相同。对于不能放在同一斜线上的这个隐含条件，可以转换，假设现在2个皇后的位置分别是(i,j)和(k,l),且i-j=k-l或者i+j=k+l时，说明2个皇后在同一斜线上。以上两个方程分别等价于i-k=j-l和i-k=l-j。由此可知，只要|i-k|=|j-l|,就表明2个皇后在同一条斜线上

代码实现

#include<iostream>
#include<vector>
#include <cmath>
using namespace std;
int n;//皇后个数
int* x;//当前解
int sum;//当前找到可行方案个数
void Print()
{
	//int **p = new int*[10];

	//for (int i = 0; i <= 10; ++i)
	//{
	//	p[i] = new int[10];
	//}

	vector<vector<int>> v(n);
	for (int i = 0; i<n; i++)
	{
		v[i].resize(n);
		for (int j = 0; j < n; j++)
		{
			v[i][j] = 0;
		}
	}

	for (int i = 0; i < n; i++)
	{
		v[i][x[i]] = 1;
	}

	cout << "结果是:" << endl;
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			cout << v[i][j] << " ";
		}
		cout << endl;

	}
	cout << endl;

}
bool place(int k)
{
	for (int j = 0; j < k; j++)
	{
		//同一斜线 同一列
		if ((abs(k - j)) == (abs(x[j] - x[k])) || (x[j] == x[k]))
			return false;
	}
	return true;
}
void backtrack(int t)
{
	if (t >= n)
	{
		Print();
		sum++;
	}
	else
	{
		for (int i = 0; i < n; i++)
		{
			x[t] = i;
			if (place(t))
				backtrack(t + 1);
		}
	}
}
int nQueen(int nn)
{
	n = nn;
	sum = 0;
	x =new int[n];
	for (int i = 0; i < n; i++)
	{
		x[i] = 0;
	}

	backtrack(0);
	return sum;
}


int main()
{

	int e = nQueen(5);
	cout << "sum = " << e << endl;
	system("pause");
}