HDU2553 N皇后问题（dfs）

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

---

 

Problem Description

在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。
你的任务是，对于给定的N，求出有多少种合法的放置方法。

---

 

Input

共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。

---

 

Output

共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。

---

 

 Sample Input

1

8

5

0

---

Sample Output

1

92

10

---

/*
 * @Descripttion: 
 * @version: 
 * @Author: ZKYAAA
 * @Date: 2020-04-16 17:56:05
 * @LastEditors: 请叫我ZK谕啊啊啊
 * @LastEditTime: 2020-04-16 18:44:34
 */
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;

//上次模拟赛，看了Alan聚聚的的题解，nice！抄了代码过来！！！！不会写那就模仿！
#define ll long long
#define ull unsigned long long
#define rint register int
#define rep(i, l, r) for (rint i = l; i <= r; i++)
#define per(i, l, r) for (rint i = l; i >= r; i--)
#define mset(s, _) memset(s, _, sizeof(s))
#define pb push_back
#define pii pair<int, int>
#define mp(a, b) make_pair(a, b)

inline int read()
{
    int x = 0, neg = 1;
    char op = getchar();
    while (!isdigit(op))
    {
        if (op == '-')
            neg = -1;
        op = getchar();
    }
    while (isdigit(op))
    {
        x = 10 * x + op - '0';
        op = getchar();
    }
    return neg * x;
}
inline void print(int x)
{
    if (x < 0)
    {
        putchar('-');
        x = -x;
    }
    if (x >= 10)
        print(x / 10);
    putchar(x % 10 + '0');
}


int n,k=0;
int col[12]={0};
bool check(int c,int r){                          //检查是否与放好得皇后冲突
    for(int i=0;i<r;i++){
       if(col[i]==c||(abs(col[i]-c)==abs(i-r)))   //不同行，不同列，不同斜线           
            return false;
    }
    return true;
}
void dfs(int r){                                  //一行一行得放皇后，第r行
    if(r==n){                                     //所有的皇后都放置好了（递归返回）
        k++;                                      //统计满足要求的数目
        return ;
    }
    for(int c=0;c<n;c++){                         //在每一列放皇后
        if(check(c,r)){                           //判断该位置是否合理
            col[r]=c;                             //在第r行第c列放置皇后
            dfs(r+1);                             //递归放下一行皇后
        }
    }
}

int main(){
    int sum[12]={0};
    for(n=0;n<=10;n++){                           //求出所有N皇后的总数，先打表，不然会TLE
        memset(col,0,sizeof(col));                //清空，准备计算下一个N皇后
        k=0;
        dfs(0);
        sum[n]=k;                                 //打表
    }
    while(cin>>n){
        if(n==0)
            return 0;
        cout<<sum[n]<<endl;
    }
    return 0;
}