hdu 2553 N皇后问题

题目：http://acm.hdu.edu.cn/showproblem.php?pid=2553

dfs的题

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<string.h>
using namespace std;
const int N=20;
int n;
int count;//多少种排列组合
int a[N];//a[i]=j代表的是在i行j列
int visit[N];//用来排除列，例：a[1]=1;上有皇后，那么visit[a[1]]=1;
int check(int i,int j)//验证第i行j列
{
    int k;
    int flag=1;
    for(k=1;k<i;k++)
    {
        if( abs(k-i) == abs(j-a[k]) )//检查是否在斜线相交
        flag=0;
    }
    return flag;
}

void dfs(int i)//i代表行数
{
     if(i>n)
     count++;
     else
     {
         int j;
         for(j=1;j<=n;j++)
         {
             if(visit[j]==0 && check(i,j))//检查位置
             {
                 a[i]=j;
                 visit[j]=1;
                 dfs(i+1);
                 visit[j]=0;
             }
         }
     }
}

int main()
{
    //freopen("in.txt","r",stdin);
    int b[15];
    memset(b,0,sizeof(b));
    int i;
    count=0;
    for(i=1;i<=10;i++)
    {
        n=i;
        memset(a,0,sizeof(a));
        memset(visit,0,sizeof(visit));
        dfs(1);//从第一行开始
        b[i]=count;
        //printf("%d ",b[i]);
        count=0;
    }
    int m;
    while(~scanf("%d",&m))//打表后输出，不然时超
    {
        if(m==0)
        break;
        printf("%d
",b[m]);
    }
    return 0;
}