C语言编程练习49：N皇后问题

#include <iostream>
#include <cstdio>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <cmath>
#include <list>
#include <cstdlib>
#include <cstring>
using namespace std;
int n,sum,vis[5][50],ans[100];
void dfs(int cur)
{
    if(cur==n+1)
    {
        sum++;
        return;
    }
    for(int i=1;i<=n;i++)
    {
        if(vis[0][i]==0&&vis[1][n+i-cur]==0&&vis[2][i+cur]==0)//vis【0】表示列。vis【1】和vis【2】表示两条对角线。i表示第几列
        {
            vis[0][i]=vis[1][n+i-cur]=vis[2][i+cur]=1;//修改标记
            dfs(cur+1);
            vis[0][i]=vis[1][n+i-cur]=vis[2][i+cur]=0;//回溯
        }
    }
}

int main()
{
    for(int i=1;i<=10;i++)
    {
        memset(vis,0,sizeof(vis));
        sum=0;
        n=i;
        dfs(1);
        ans[i]=sum;
    }
    int t;
    while(cin>>t)
    {
        if(t==0)
        {
            break;
        }
        cout<<ans[t]<<endl;
    }
    return 0;
}