HDU 2553(N皇后)（DFS）

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

i表示行，map[i]表示列，然后用DFS遍历回溯

可以参考这篇文章：

http://blog.csdn.net/cambridgeacm/article/details/7703739

#include <iostream>
#include <string>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <stack>
using namespace std;

#define MEM(a,b) memset(a,b,sizeof(a))

bool used[15];
//i表示行，map[i]表示列
int map[15];
int sol[15];

int ans,n;


void dfs(int k)
{
          int i,j;

          //判断末尾
          if(k==n+1)
          {
                    ans++;
                    return;
          }

          for(i = 1;i<=n;i++)
          {
                    if(!used[i])
                    {
                              map[k] = i;
                              bool flag = true;

                              //是否在一条斜线上
                              for(j=1;j<=k-1;j++)
                              {
                                        if((map[k]-map[j])==(k-j) || (map[k]-map[j])==(j-k))
                                        {
                                                 flag = false;
                                                 break;
                                        }
                              }
                              if(flag)
                              {
                                        used[i] = true;
                                        dfs(k+1);
                                        //没找到，释放used
                                        //回溯
                                        used[i] = false;
                              }
                    }
          }
}


int main()
{
          int t,i;
          //保存是因为后台可能有重复的数据，单次来说效率是一样的
          for(i = 1;i<=10;i++)
          {
                    MEM(used,false);
        MEM(map,0);
        n = i;
        ans = 0;
        dfs(1);
        sol[i] = ans;
          }

    while (~scanf("%d",&n) && n)
    {
                    printf("%d
",sol[n]);
    }
    return 0;
}