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

最新推荐文章于 2021-03-23 12:41:15 发布

Grit_ICPC

最新推荐文章于 2021-03-23 12:41:15 发布

阅读量441

点赞数

分类专栏： BFS&&DFS 文章标签： dfs

本文链接：https://blog.csdn.net/Grit_ICPC/article/details/51353452

版权

BFS&&DFS 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

大意：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;
}