【1219】八皇后问题

1.最开始参考算法笔记，采用全排列的思想，很不幸，超时了，有些确定无效的放置方式就不该再进行dfs搜索了。
只得了61分，代码如下：
#include <bits/stdc++.h> using namespace std; //在全排列基础上求解八皇后问题 int n; int ans=0; bool vis[20]; int a[20]; void dfs(int index){ if(index==n+1){ int flag=0; for(int i=1;i<=n;i++){ for(int j=i+1;j<=n;j++){ //每一行和下面所有行比 if((j-i)==abs(a[i]-a[j])){ flag=1; } } } if(flag==0){ ans++; if(ans<=3){ for(int i=1;i<=n;i++) cout<<a[i]<<" "; cout<<endl;}}} for(int i=1;i<=n;i++){ if(vis[i]==false){ vis[i]=true; a[index]=i; dfs(index+1); vis[i]=false; } } } int main(){ cin>>n; dfs(1); cout<<ans; }
2.采用回溯法，可以ac
#include <bits/stdc++.h> using namespace std; //在全排列基础上求解八皇后问题 int n; int ans=0; bool vis[13]; int a[13]; void dfs(int index){ if(index==n+1){ ans++; if(ans<=3){ for(int i=1;i<=n;i++) cout<<a[i]<<" "; cout<<endl;} } //dfs到达终点的条件，这个是会的 for(int i=1;i<=n;i++){ if(vis[i]==false){ int flag=0; for(int j=1;j<index;j++){ if(abs(index-j)==abs(i-a[j])) //判断和之前的皇后是否在一个对角线上 flag=1; } //index 现在要放置的横坐标位置，横坐标要放置的数的值 if(flag==0){ vis[i]=true; a[index]=i; dfs(index+1); vis[i]=false; } } } } int main(){ cin>>n; dfs(1); cout<<ans; }