HDU 2553 N皇后问题 （DFS+回溯）

大意：n皇后在n*n的棋盘上，要任意两皇后不能同行列并且在与x成45度的地方也不能有皇后。

思路：主要是判断与对角线平行的地方的皇后是否有冲突，我们可以用一个mp【】数组下标为行值为列。两mp[]值是否相等||mp[X]-X==mp[Y]-Y||mp[X]+X==mp[Y]+Y

#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<stack>
#include<iostream>
#include<cstring>
#include<algorithm>
#define LL int
#define inf 0x3f3f3f3f
#define eps 1e-8
#include<vector>
#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1
using namespace std;
int ans,n,cnt[15],mp[15];
bool vis[15];
void DFS(int step){
if(step == n){
ans++; return ;
}
for(int i = 0;i < n;++ i){
if(!vis[i]){
mp[step] = i;
bool bj = false;
for(int j = 0;j < step;++ j){
if( (mp[step] == mp[j])||(mp[step]-step== mp[j] - j)||(mp[step]+step== mp[j]+j) ){
bj = true;break;
}
}
if(!bj){
vis[i] = true;
DFS(step+1);
vis[i] = false;
}
}
}
}

int main(){
int m,i,j;
for(i = 1;i <= 11;++ i){
memset(vis,false,sizeof(vis));
memset(mp,0,sizeof(mp));
ans = 0;
DFS(0);
n = i;
cnt[i] = ans;
}
while(~scanf("%d",&n)&&n){
printf("%d\n",cnt[n+1]);
}
return 0;
}