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;
}
这道题数据比较小,枚举就能过,超时的原因还没找出来。。。
浙公网安备 33010602011771号