HDU - 2553N皇后问题(启发式搜索---回溯是关键)

题目链接:https://cn.vjudge.net/contest/311091#problem/F
在N*N的方格棋盘放置了N个皇后，使得它们不相互攻击（即任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上。
你的任务是，对于给定的N，求出有多少种合法的放置方法。
Input
共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。
Output
共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。
Sample Input

1
8
5
0

Sample Output

1
92
10

解析：
1.解题的关键是，如何处理任意2个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成45角的斜线上这三个限制条件。
2.N<=10，这样可以打表处理，避免重复操作。
3.找找规律。

黄线和红线都是与棋盘边框成45角的斜线。
对于红线，线上各点的坐标，x+y是定值。
对于黄线，线上各点的坐标，x-y是定值。
用book数组标记就能解决问题。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[10],s,sum;
int book1[20],book2[20],book3[20];
void dfs(int n)//第n行，同时表示放了几个皇后
{
    if(n==s+1){
        sum++;
        return;
    }
    for(int i=1;i<=s;i++){//n-i+s是为了避免负数
        if(book1[i]==0&&book2[n-i+s]==0&&book3[n+i]==0){
            book1[i]=1,book2[n-i+s]=1,book3[n+i]=1;
            dfs(n+1);
            book1[i]=0,book2[n-i+s]=0,book3[n+i]=0;
        }
    }
}
int main()
{
    memset(a,0,sizeof(a));
    a[1]=1;
    for(int i=2; i<=10; i++){
        memset(book1,0,sizeof(book1));
        memset(book2,0,sizeof(book2));
        memset(book3,0,sizeof(book3));
        s=i;//dfs中结束条件会用到
        sum=0;
        dfs(1);
        a[i]=sum;
    }
    int x;
    while(~scanf("%d",&x)&&x)
        printf("%d\n",a[x]);
    return 0;
}