【回溯法】n皇后问题-DFS

题目

在n * n的棋盘上放置彼此不受攻击的n个皇后。按照国际象棋的规则，皇后可以攻击与之在同一行，同一列，同一斜线上的棋子。现在在n * n的棋盘上放置n个皇后，使彼此不受攻击。

问题分析

在放置皇后的过程中，我们要考虑三个方面，不能同行，不能同列，不能同对角线。那么就可以从行出发，即每个皇后放在一行中，这样只需考虑与已放置皇后不同列，不同对角线即可。也可以从列出发，每个皇后先是放在不同的列上，然后再考虑不同行和不同对角线。本题从行出发。

算法

本题采用回溯法，也可以叫深度优先搜索（DFS）。本题的解空间是一个m叉树，就是除叶子结点外，每个结点都有m个分支。使用递归的方法一个皇后一个皇后地放置，利用约束条件，就是已放置皇后不同列且不同对角线，来确定能否继续向下扩展。如果能，就继续下一个皇后的放置；如果不能，就回溯到最近的活结点，然后扩展其他分支。

代码实现

#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=105;

int n;//皇后个数
int x[maxn];//存储皇后的位置
int cnt;//放法个数

bool place(int t)
{
    bool flag=true;
    for(int j=1; j<t; ++j)
    {
        if(x[j]==x[t]||t-j==fabs(x[j]-x[t]))//注意作差顺序
        {
            flag=false;
            break;
        }
    }
    return flag;
}

void dfs(int t)
{
    if(t>n)
    {
        cnt++;
        cout<<"第"<<cnt<<"种：";
        for(int i=1; i<=n; ++i)
            cout<<x[i]<<" ";
        cout<<endl;
        return ;
    }
    else
    {
        for(int i=1; i<=n; ++i) //i不能定义为全局变量，递归调用的时候会有问题
        {
            x[t]=i;
            if(place(t))
                dfs(t+1);
        }
    }
}

int main()
{
    cout<<"请输入皇后个数：";
    cin>>n;
    memset(x,0,sizeof(x));
    cnt=0;
    dfs(1);
    cout<<"一共有"<<cnt<<"种放法。"<<endl;
    return 0;
}

参考文献：《趣学算法》