C/C++ 算法分析与设计：搜索（N皇后问题）

题目描述

         检查一个如下的6 x 6的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行，每列，每条对角线(包括两条主对角线的所有对角线)上都至多有一个棋子。

         例如，可以用序列2 4 6 1 3 5来描述一个布局，第i个数字表示在第i行的相应位置有一个棋子。

         请遍一个程序找出所有跳棋放置的解。并把它们以上面的序列方法输出。解按字典顺序排列。请输出前3个解。最后一行是解的总个数。 

 

输入

使一个数字N (6 <= N <= 13) 表示棋盘是N x N大小的。

输出

前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

样例输入

6

样例输出

2 4 6 1 3 5 
3 6 2 5 1 4 
4 1 5 2 6 3 
4

#include<iostream>
using namespace std;
int a[20][20] = { 0 };
int flag[20] = { 0 };
int n,c=0,b=0;
bool f(int x, int y)
{
	int i, k;
	for (i = x, k = y; k >= 1; i--, k--)if (a[i][k] == 1)return false;//判断左上方
	for (i = x, k = y; k <= n; i--,k++ )if (a[i][k] == 1)return false;//判断右下方
	for (i = x, k = y; i >= 1; i--)if (a[i][k] == 1)return false;//判断上方
	return true;
}
void dns(int k)//k代表行数
{
	if (k == n+1)
	{
		b++;
		if (b <= 3)
		for (int i = 1; i <= n; i++)
		{
			cout << flag[i] << ' ';
			if (i == n)cout << endl;
		}
		
	}
	else
	{
		for (int i = 1; i <= n; i++)//遍历所有的列数
		{
			if (a[k][i] == 0 && f(k, i))
			{
				flag[k] = i;
				a[k][i] = 1;
				dns(k + 1);
				a[k][i] = 0;
			}
		}
	}
}
int main()
{
	while (cin >> n)
	{
		c = 0;
		b = 0;
		dns(1);

		cout << b << endl;
	}
	return 0;
}