回溯算法——N皇后问题

回溯算法

回溯算法的理论基础

回溯法的基本思想

问题描述

在n×n格的棋盘上放置彼此不受攻击的n个皇后。
按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。n皇后问题等价于在n×n格的棋盘上放置n个皇后，任何两个皇后不放在同一行或同一列或同一斜线上。
编程要求：找出一个n×n格的棋盘上放置n个皇后并使其不能互相攻击的所有方案。

输入输出样例

分析

由于棋盘的每列只有一个皇后，所以可以用一维向量X( x1, x2, …, xn)，其中xi∈{1, 2, …, n}，表示第i列皇后所在的行x[i]，即解空间的每个结点都有n个儿子，因此解空间的大小为nn，这是一棵子集树。

代码

#include<bits/stdc++.h>
#define NUM 20
using namespace std;
int  n;    //棋盘的大小
int x[NUM];//解向量
int sum;   //当前已经找到的可行方案数
inline bool Place(int t)//形参t是回溯的深度
{
    int i;
    for (i=1; i<t; i++)
    {
        if ((abs(t-i) == abs(x[i]-x[t])) || (x[i] == x[t]))
            return false;
    }
    return true;
}
//形参t是回溯的深度，从1开始
void Backtrack(int t)
{
    int i;
    //到达叶子结点，获得一个可行方案。累计总数，并输出该方案
    if(t>n)
    {
        sum++;  //是全局变量
        for(i=1;i<=n;i++)
            cout<<x[i]<<" ";
        cout<<endl;
    }
    else
        for(i=1;i<=n;i++)
        {
            x[t]=i;
            if(Place(t))
                Backtrack(t+1);
        }
}
int main()
{
    cin>>n;
    Backtrack(1);
    cout<<"Total="<<sum<<endl;
    return 0;
}