代码改变世界

N皇后问题(dfs)

2019-07-24 20:22  木木王韦  阅读(111)  评论(0)    收藏  举报

N皇后问题

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

Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input
1
8
5
0
Sample Output
1
92
10

这是一个超时的代码:


#include<iostream>
#include<algorithm>
using namespace std;

int vis[11];
int ans;
int n;

void dfs(int x){
	if(x==n) ans++;
	for(int i=0;i<n;i++){
		int flag=0;
		vis[x]=i;
		for(int j=0;j<x;j++)
         {
		if(i==vis[j]||i+x==j+vis[j]||i-x==vis[j]-j)
             {
                 flag=1;
                 break;
             }
         }
         if(!flag) dfs(x+1);
	}
}


int main(){
	while(cin>>n&&n){
		ans=0;
		dfs(0);
		cout<<ans<<endl;
	}
	
	return  0;
}

Comment in Markdown
Preview:

这样写也超。。。

#include<iostream>
#include<algorithm>
using namespace std;

int vis[11];
int ans;
int i,n;

void dfs(int x){
	if(x==n) ans++;
	
	for(int i=0;i<n;i++){
		int flag=0;
		vis[x]=i;
		for(int j=0;j<x;j++)
         {
		if((i==vis[j]||i+x==j+vis[j]||i-x==vis[j]-j)&&!flag)
             {
                 flag=1;
                 break;
             }
         }
         if(!flag) dfs(x+1);
	}
}


int main(){
	int num[11];
	for(i=1;i<11;i++){
		ans=0;
		dfs(0);
		num[i]=ans;
	}
	
	
	while(cin>>n&&n){
		ans=0;
		dfs(0);
		cout<<ans<<endl;
	}
	
	return  0;
}

这道题数据比较小,枚举就能过,超时的原因还没找出来。。。