HDU 2553 N皇后问题

N皇后问题

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 14958    Accepted Submission(s): 6813

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
   
   
   
   

分析：

用回溯法写的皇后问题，但是直接交回TLE，所以用数组先存一下，输入一个n直接输出数组内的值就OK。

#include <stdio.h>
#include <math.h>
#include <string.h>
int a[15],b[15];
int queue[15];
int max;

int judge(int k,int t){
	int i;
	for(i=1;i<t;i++)
	{
		if(fabs(a[i]-k)==fabs(i-t))		//判断是否在同一条斜线上 
			return 0;
	}
	return 1;
}

void dfs(int t,int n)
{
	int i;
	if(t==n+1)
	{
		max++;
		return ;
	}
	
	for(i=1;i<=n;i++)
	{
		if(!b[i] && judge(i,t))	//剪枝 
		{
			b[i]=1;
			a[t]=i;
			dfs(t+1,n);
			b[i]=0;
		}
	}
}

int main()
{
	int n;
	int i,j;
	memset(a,0,sizeof(a));
	for(i=1;i<=10;i++)
	{
		max=0;
		memset(b,0,sizeof(b));
		dfs(1,i);
		queue[i]=max;
	}
	while(scanf("%d",&n),n)
		printf("%d\n",queue[n]);
	
	return 0;
}