【算法】n皇后问题

描述

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

输入输出

输入

5

输出

1 3 5 2 4
1 4 2 5 3
2 4 1 3 5
2 5 3 1 4
3 1 4 2 5
3 5 2 4 1
4 1 3 5 2
4 2 5 3 1
5 2 4 1 3
5 3 1 4 2
Total= 10

分析

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

代码

#include<bits/stdc++.h>
using namespace std;

#define NUM 20
int n;
int x[NUM];
int sum;

bool Place(int 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;
}

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()
{
    while (cin>>n)
    {
        sum = 0;
        Backtrack(1);
        cout<<"Total="<<sum<<endl;

    }
    return 0;
}