DFS与八皇后问题

DFS与八皇后问题（P1219）

• 题目描述

  一个如下的 6×66 \times 66×6 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

  

  上面的布局可以用序列 2 4 6 1 3 5来描述，第 iii 个数字表示在第 i 行的相应位置有一个棋子，如下：

  行号 1 2 3 4 5 6

  列号 2 4 6 1 3 5

  这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
  并把它们以上面的序列方法输出，解按字典顺序排列。
  请输出前 3 个解。最后一行是解的总个数。

  输入格式

  一行一个正整数 n，表示棋盘是 n×n 大小的。

  输出格式

  前三行为前三个解，每个解的两个数字之间用一个空格隔开。第四行只有一个数字，表示解的总数。

  输入输出样例

  **输入 **

  6
  

  **输出 **

  2 4 6 1 3 5
  3 6 2 5 1 4
  4 1 5 2 6 3
  4
  

  说明/提示

  【数据范围】
  对于 100%的数据，6≤n≤13

解题思路

• 八皇后问题要求每行每列，每条对角线及其平行线上只能有一个棋子，那么即可使用四个一维数组来表示上述每个位置是否有棋子。

  然后使用DFS搜索所有点（注意要回溯，因为要找出所有结果），满足条件时输出结果（因为DFS是按序搜索每一个点，所以题中的字典序不需要考虑，直接输出前三个就行）。

代码实现

#include<iostream>
#include<algorithm>
#include<string>
#include<map>
#include<vector>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=100005;
int n;
int a[100],b[100],c[100],d[100];
int sum=0;

void DFS(int i){
    if(i>n){
        if(sum<=2){
            for(int j=1;j<=n;j++){
                cout<<a[j]<<" ";
            }
            cout<<endl;
        }+++
        sum++;
        return ;
    }
    else{
        for(int j=1;j<=n;j++){
            if((!b[j])&&(!c[i+j])&&(!d[i-j+n])){
                a[i]=j;
                b[j]=1;
                c[i+j]=1;
                d[i-j+n]=1;
                DFS(i+1);
                b[j]=0;
                c[i+j]=0;
                d[i-j+n]=0;
            }
        }
    }
}

int main(){
    cin>>n;
    DFS(1);
    cout<<sum<<endl;


    system("pause");
    return 0;
}