杭电hdu N皇后问题

N皇后问题

Time Limit : 2000/1000ms (Java/Other)   Memory Limit : 32768/32768K (Java/Other)

Total Submission(s) : 11   Accepted Submission(s) : 4

Font: Times New Roman | Verdana | Georgia

Font Size: ← →

Problem Description

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

Input

共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

Output

共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

Sample Input

1
8
5
0

Sample Output

1
92
10

利用回溯法解决

程序首先对N行中的每一行进行探测，寻找该行中可以放置皇后的位置，具体方法是对该行的每一列进行探测，看是否可以放置皇后，如果可以，则在该列放置一个皇后，然后继续探测下一行的皇后位置。如果已经探测完所有的列都没有找到可以放置皇后的列，此时就应该回溯，把上一行皇后的位置往后移一列，如果上一行皇后移动后也找不到位置，则继续回溯直至某一行找到皇后的位置或回溯到第一行，如果第一行皇后也无法找到可以放置皇后的位置，则说明已经找到所有的解程序终止。如果该行已经是最后一行，则探测完该行后，如果找到放置皇后的位置，则说明找到一个结果，打印出来。但是此时并不能再此处结束程序，因为我们要找的是所有N皇后问题所有的解，此时应该清除该行的皇后，从当前放置皇后列数的下一列继续探测。

完整代码如下，注释基本可以看懂了

#include<iostream>
using namespace std;
int map[11];
bool place(int row,int col,int k)
{
	for(int i=1;i<k;i++)
	{
		if(col==map[i]||abs(row-i)==abs(col-map[i]))
			return false;
	}
	return true;
}
int main()
{
	int n;
	while(cin>>n)
	{
		if(n==0)
			return 0;
		memset(map,0,sizeof(map));
		map[1]=1;
		int k=1,count=0;
		while(k>0)
		{
			if(k<=n && map[k]<=n)
			{
				if(place(k,map[k],k))
				{
					k++;
					map[k]=1;
				}
				else
				{
					map[k]++;
				}
			}
			else
			{
				if(k>n)
					count++;     
				k--;
				map[k]++;
			}
		}
		cout<<count<<endl;
	}
	return 0;
}