八皇后问题

问题描述：
在N×N（4<=N<=13）格的国际象棋上摆放八个皇后，使其不能互相攻击，即任意两个皇后都不能处于同一行、同一列或同一斜线上，问有多少种摆法。

相关联系：
洛谷P1219：链接

代码（我的编译环境为vs2017）：
对于其中主对角线及副对角线的问题，这张图可以帮助理解。

#include "stdafx.h"
#include <stdlib.h>
#include <iomanip>
#include <iostream>
#include <ctime>
#include <cstring>
#include <stdio.h>
#include <vector>
using namespace std;

void search(int N, int k, int& count, vector<int>&vec, bool vis[3][27])
{
    if (k == N)//能到达这里的k都是满足条件的方案
    {
        if (count < 3)//只输出前三个方案
        {
            for (int i = 0; i < N - 1; i++)
                cout << vec[i] << " ";
            cout << vec[N - 1] << endl;
        }
        count++;
    }
    else
    {
        for (int i = 1; i <= N; i++)
        {
            //因为皇后是逐行放入的，所以不用担心他们在同一行

            /*
            !vis[0][i]判断i要放入的列有没有皇后
            !vis[1][k + 1 + i] 判断i要放入的主对角线有没有皇后
            !vis[2][k + 1 - i + N]判断i要放入的副对角线有没有皇后
            */
                if (!vis[0][i]&&!vis[1][k + 1 + i] && !vis[2][k + 1 - i + N])
                {
                    vis[0][i]=vis[1][k + 1 + i] = vis[2][k + 1 - i + N] = 1;
                    //i放入后，将i所在的列、主对角线、副对角线标记为有皇后

                    vec.push_back(i);//将i放入容器中
                    search(N, k + 1, count, vec, vis);//进入递归
                    vec.pop_back();//递归结束后从容器中取出i

                    vis[0][i]=vis[1][k + 1 + i] = vis[2][k + 1 - i + N] = 0;
                    //递归结束后取消对i所在的列对角线、副对角线的标记
                }
        }
    }
}

int main()
{
    int N, count=0;
    cin >> N;
    vector<int>vec;
    bool vis[3][27] = { 0 };
    search(N, 0, count, vec, vis);
    cout << count << endl;
    return 0;
}