首页 > > 程序设计 > C/C++ >

hdu-2553 N皇后问题（搜索题）

2018-08-21 05:28:24来源：博客园阅读 ()

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

Input共有若干行，每行一个正整数N≤10，表示棋盘和皇后的数量；如果N=0，表示结束。Output共有若干行，每行一个正整数，表示对应输入行的皇后的不同放置数量。Sample Input

Sample Output

1
92
10


题解：可以说是入门的dfs了，直接用数组的i来储存行就不用考虑行的重复了， 只需要进行对列和对角线的判断就好了，先打个表，不然可能会超时哦

#include<iostream>
#include<algorithm>
#include<cstring>
#include<sstream>
#include<cmath>
#include<cstdlib>
#include<queue>
using namespace std;
#define PI 3.14159265358979323846264338327950

int map[12],a[12],n,sum;

int check(int x)
{
    for(int i=0;i<x;i++)
    {
        if(map[i]==map[x]||abs(map[i]-map[x])==abs(i-x))
            return 0;
    }
            return 1;
}
void dfs(int x)
{
    if(x==n)
        sum++;
    else
    {
        for(int j=1;j<=n;j++)
        {
            map[x]=j;
            if(check(x))
                dfs(x+1);
        }
    }
}

int main()
{
    for(int i=1;i<11;i++)
    {
        sum=0;
        n=i;
        dfs(0);
        a[i]=sum;
    }
    int n;
    while(scanf("%d",&n) && n)
    {
        printf("%d\n",a[n]);
    }
}