n皇后问题 —— 回溯法

• 题目
  • 在n*n格的棋盘上摆放n个皇后，使其不能互相攻击，即任意两个皇后都不能处于同一行、同一列或同一对角线上，问有多少种摆法？
• c++代码

#include <iostream>
#include <math.h>

using namespace std;

int p;  //存储皇后的个数
int sum = 0;     //存储可能性的个数
int a[100];   //下标表示行，值表示皇后所在列
int flag = 0;  //位置是否合法的标记


void check(int n) {  //判断皇后所在位置是否合法
    flag = 0;
    for (int i = 0; i < n; i++) {    //和皇后n之前的所有皇后进行比较，看位置是否合法
        if (a[i] == a[n] || abs(a[i] - a[n]) == abs(i - n)) {      //不合法的情况有两种，满足其中任意一个就是不合法
           /* 比较是否合法的的方式：
            1.因为是一行一行进行的，所以不存在同行的情况
            2.当a[i] == a[n]，即两个皇后同列，位置不合法
            3.当n皇后和i皇后在同一条对角线上，即斜率为+1或-1，经过变形就是a[i]-a[n]的绝对值 == i-n ，
                即： | a[i]-a[n] | = | i-n | ；
                在c++中，用abs函数求绝对值，使用头文件#include <math.h>
            */
            flag = -1;  //如果不合法，n皇后的位置标记为-1
            break;  //不合法，跳出循环，进行下一个函数
        }
    }
}

void queen(int n) {    //放置皇后
    if (n == p) {   //如果判断到p皇后，就意味着所有的皇后都已经摆放好了：因为在数组中皇后的下标是“0 ~ p-1" 

        sum++;    //如果最后一个皇后也满足条件，那sum+1,即摆放方法+1

        return;  //结束这个循环,因为p个皇后都已经摆放好了
    }

    for (int j = 0; j < p; j++) {  //在一行中依次选择列，进行合法性判断

        a[n] = j;   //从第一列开始摆放皇后n

            check(n);   //进行合法性判断

            if (flag == 0) {  //如果位置合法，摆放下一个皇后，直至最后一个皇后

                queen(n + 1);   //递归判断下一个皇后的位置

            }


            //如果位置不合法，重新选择n皇后所在的列
    }
}



int main()
{
    cout << "输入皇后的个数：" << endl;
    cin >> p;
    queen(0);
    cout << "\n";
    cout << p << "皇后有" << sum << "种摆放方式" << endl;
    return 0;
}