HDU 2553 N皇后问题（dfs）

Description
在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。你的任务是，对于给定的N，求出有多少种合法的放置方法
Input
共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。
Output
共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。
Sample Input
1
8
5
0
Sample Output
1
92
10
Solution
简单dfs
Code

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,res,map[12],ans[12],vis[12];
void dfs(int k)
{
    if(k==n+1)//合法的放置方法 
    {
        res++;
        return ;
    }
    for(int i=1;i<=n;i++)
        if(!vis[i])//各行棋子不能在同一列 
        {
            map[k]=i;
            int flag=1;
            for(int j=1;j<k;j++)
                if((map[k]-map[j])==(k-j)||(map[k]-map[j])==(j-k))//两个皇后不能同在同一斜线上 
                {
                    flag=0;
                    break;
                }
            if(flag)
            {
                vis[i]=1;
                dfs(k+1);
                vis[i]=0;//回溯 
            }
        }
}
int main()
{
    memset(ans,0,sizeof(ans));//初始化 
    for(int i=1;i<=10;i++)//打表 
    {
        n=i;
        res=0;
        memset(map,0,sizeof(map));
        memset(vis,0,sizeof(vis));
        dfs(1);
        ans[i]=res;//记录答案 
    }
    while(scanf("%d",&n),n)
        printf("%d\n",ans[n]);
    return 0;
}