USACO training course Checker Challenge N皇后 /// oj10125

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<cstring>
#include<math.h>
using namespace std;
int n,cnt;
int ans[100],vis[5][50];
void DFS(int m)
{
    if(m==n+1)
    {
        cnt++;
        if(cnt<=3)
        {
            for(int i=1;i<n;i++)
                printf("%d ",ans[i]);
            printf("%d\n",ans[n]);
        }
        return ;
    }
    for(int i=1;i<=n;i++)
    {
        if(!vis[0][i]&&!vis[1][m+i]&&!vis[2][m-i+n])
        {
            vis[0][i]=vis[1][m+i]=vis[2][m-i+n]=1;
            ans[m]=i; DFS(m+1);
            vis[0][i]=vis[1][m+i]=vis[2][m-i+n]=0;
        }
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(ans,0,sizeof(ans));
        cnt=0; DFS(1);
        printf("%d\n",cnt);
    }

    return 0;
}