2017.5.3 3.n皇后问题

题目描述

在N*N(N<=10)的棋盘上放N个皇后，使得她们不能相互攻击。两个皇后能相互攻击当且仅当它们在同一行，或者同一列，或者同一条对角线上。 找出一共有多少种放置方法。

输入

第一行输入N。

输出

输出方案总数。

样例输入

4

样例输出

2

数据范围限制

N<=10

之前用暴力DFS写过一次，今天又尝试学了一下剪枝。

问题的解可以用一个n元向量X=（x1,x2,.....xn）表示，即第n个皇后放在第i行第xi列上。 两个皇后不能放在同一列上，xi≠ xj;若两个皇后的摆放位置分别是（i,xi）和（j,xj），在棋盘上斜率为-1的斜线上，满足条件i-j=xi-xj;在棋盘上斜率为1的斜线上，满足条件i+j=xi+xj; 综合两种情况，由于两个皇后不能位于同一斜线上，所以|i-xi|≠ |j-xj|。

#include<cstdio>

int n,ans; 
int a[15];

void DFS(int u)
{
    if(u>n)
        {
            ++ans;
            return;
        }
    for(int i=1;i<=n;i++)
        {
            bool law=1;
            for(int j=1;j<u&&law;j++)
                if(a[j]==i||a[j]+j==i+u||a[j]-j==i-u)
                    law=0;
            if(!law) continue;
            a[u]=i;
            DFS(u+1);
        }
}

int main()
{
    scanf("%d",&n);
    ans=0;
    DFS(1);
    printf("%d
",ans);
    return 0;
}