N皇后问题

问题描述：

一般地，N皇后问题描述如下：

在大小为N×N的棋盘上摆放N个皇后，使其两两之间不能互相攻击，即任意两个皇后都不能处于棋盘的同一行、同一列或同一斜线上，求出满足条件的所有棋局及局面总数。

特殊地，当N=8时为著名的以国际象棋棋盘为背景的8皇后问题，易解得符合条件的局面总数为92。

代码

#include<iostream>
using namespace std;
int a[100][100];
int n, vis1[100]/*标记列*/, vis2[100]/*标记左上到右下*/, vis3[100]/*标记右上到左下*/;
void vis(int i, int j, bool flag)
{
    a[i][j] = flag;
    vis1[j] = flag;
    vis2[i - j + n] = flag;
    vis3[i + j] = flag;
}
int ans = 1;
void dfs(int i)  //状态为当前的行数
{
    if (i == n)  //递归结束条件：当递归到最后一行即n-1行时仍存在目标，此时进入第n行结束
    {
        printf("NO.%d\n", ans++);
        for(int i=0;i<n;i++){
            for (int j = 0; j < n; j++)
                printf("%d ", a[i][j]);
            printf("\n");
            }
        return;
    }
    for (int j = 0; j < n; j++)  
    {
        if (!vis1[j] && !vis2[i - j + n] && !vis3[i + j])
        {
            vis(i, j, 1);
            dfs(i + 1);
            vis(i, j, 0); //回溯
        }
    }
}
int main()
{
    cin >> n;
    dfs(0);  
    return 0;
}