DFS(4):经典N皇后问题

1-problem:

会下国际象棋的人都很清楚：皇后可以在横、竖、斜线上不限步数地吃掉其他棋子。如何将8个皇后放在棋盘上（有8 * 8个方格），使它们谁也不能被吃掉！这就是著名的八皇后问题。

2-input:

一个整数n（ 1 < = n < = 10 )

3-output:

每行输出对应一种方案，按字典序输出所有方案。每种方案顺序输出皇后所在的列号，相邻两数之间用空格隔开。如果一组可行方案都没有，输出“no solute!”

4-eg:

样例输入

4

样例输出

2 4 1 3
3 1 4 2

5-solution:

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

vector<vector<int>> results;
vector<int> tmp;
int n;//输入的N皇后
int k=0;//表示有多少种方法
void dfs(int dmp,vector<bool>& vis,vector<int>& p)
{
     if(dmp == n)
     {
         for(int i=0;i<n;i++)
         {
             cout<<p[i]+1<<" ";
         }
         cout<<endl;
         ++k;
     }
     else
     {
         for(int i=0;i<n;i++)//以列为索引值遍历
         {
             if(!vis[i])//同列已经排除
            {
               int flag=1;//判断是否该列可以放置
               for(int pre = 0;pre<dmp;pre++)//pre<dmp排除同行
               {
                    if(abs(dmp-pre) == abs(i-p[pre]))//对角线则行差=列差
                    {
                        flag=0;
                        break;
                    }
               }
               if(flag==1)
               {
                   p[dmp]=i;//该行对应的列号
                   vis[i]=1;
                   dfs(dmp+1,vis,p);
                   vis[i]=0;
               }
            }
         }
     }
}
int main()
{
    cin>>n;
    vector<bool> vis(n,0);
    vector<int> p(n,0);
    dfs(0,vis,p);//以行为索引值遍历
    if(k==0)
       cout<<"no solute!"<<endl;
    return 0;
}