试题:八皇后问题

问题背景：

八皇后问题是一个古老而著名的问题，是回溯算法的典型例题。该问题是十九世纪著名的数学家高斯1850年提出：在8X8格的国际象棋上摆放八个皇后，使其不能互相攻击，即任意两个皇后都不能处于同一行、同一列或同一斜线上，问有多少种摆法。
高斯认为有76种方案。1854年在柏林的象棋杂志上不同的作者发表了40种不同的解，后来有人用图论的方法解出92种结果。

#include <iostream>
#include <memory.h>
using namespace std;

int count;    // 记录八皇后正确安置的方法个数
bool row[8];  // row[i]为true, 说明第i+1行有一个皇后
bool col[8];  // col[j]为true, 说明第j+1列有一个皇后
bool l2r[15]; // l2r[i-j+7]为true, 说明第i+1行第j+1列所在位置上的对角线上有一个皇后
bool r2l[15]; // r2l[i+j]为true, 说明第i+1行第j+1列所在位置上的反对角线上有一个皇后
char pos[8];  // pos[i]记录第i+1行的皇后所在的列数

void init()
{
    count = 0;
    memset(&row, 0, sizeof(bool) * 8);
    memset(&col, 0, sizeof(bool) * 8);
    memset(&l2r, 0, sizeof(bool) * 15);
    memset(&r2l, 0, sizeof(bool) * 15);
    memset(&pos, 0, sizeof(char) * 8);
}

void mark(int i, int j, bool occupy)
{
    row[i] = occupy;
    col[j] = occupy;
    l2r[i - j + 7] = occupy;
    r2l[i + j] = occupy;
}

bool check(int i, int j)
{
    return (!row[i] && !col[j] && !l2r[i - j + 7] && !r2l[i + j]);
}

void draw()
{
    cout << "                 /--- " << count << " ---\\" << endl;
    for (int k = 0; k < 8; k++) {
        cout << "(" << k << "," << (int)pos[k] << ") ";
    }

    cout << endl << "        --- --- --- --- --- --- --- --- " << endl;
    for (int i = 0; i < 8; i++) {
        cout << "       |";
        for (int j = 0; j < 8; j++) {
            if (j != pos[i]) {
                cout << "   |";
            }
            else {
                cout << " * |";
            }
        }
        cout << endl << "        --- --- --- --- --- --- --- --- " << endl;
    }
    cout << endl;
}

void search(int i)
{
    for (int j = 0; j < 8; j++) {
        if (check(i, j)) {
            pos[i] = j;
            if (7 == i) {
                ++count;
                draw();
            }
            else {
                mark(i, j, true);
                search(i + 1);
                mark(i, j, false);
            }
        }
    }
}

int main()
{
    init();
    search(0);
    cout << endl << "      total number of feasible scheme = " << count << endl;
    return 0;
}

代码中的l2r和r2l两个数组的原理是看到有道百科上的, 这使得判断对角线与反对角线上是否可行变得极为容易, 否则我也就没心情去做这件事了, 第一个想到这个东西的人还是挺牛的, 我呢, 只是知道对应线上i-j和反对角线上i+j是有规律的, 麻木动手, 确实很不理智