N皇后放置方法种数

#include <iostream>
#include <cstdio>
using namespace std;
int a[11],n,temp;
void dfs(int k)
{
   int i,j,flag;
    if(k==n+1)
    {
        temp++;
        return ;
    }
    else
    {
        for(i=1;i<=n;++i)
        {
            a[k]=i;
            flag=1;
            for(j=1;j<k;++j)
            {
                if(a[j]==i||i-k==a[j]-j||i+k==a[j]+j)
                  {
                    flag=0;
                    break;
                  }
            }
            if(flag)
                dfs(k+1);
        }
    }

}
int main()
{
    int b[11],m;
    for(n=1;n<=10;++n)
    {
        temp=0;
        dfs(1);
        b[n]=temp;
    }
    while(scanf("%d",&m),m)
    {
        printf("%d\n",b[m]);
    }
    return 0;
}