八皇后(回溯法)

八皇后问题是一个十分经典的回溯问题,我们将其分析一下可以发现,八皇后问题其实是一个全排列问题,行号代表排列的下标号,列号代表1~n的数字。求得1~n的全排列,回溯找到其中符合要求的情况,这样最多找8!次,即40320次,很小的一个数据量了。在处理是否符合要求时,要注意是否是不同列、不同主对角线、不同副对角线,不需要判断是否是不同行,因为是列的全排列,肯定在不同的行。难点在于对角线的判断,主对角线:y1-x1 == y2-x2。副对角线:y1+x1 == y2+x2。

八皇后优化效率的方式是加了个辅助的二维数组vis[2][10005]用于在循环内容执行前判断是否符合要求。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <sstream>
#include <cmath>
#include <queue>
using namespace std;
const int maxn = 100005;
int n,tot;
int c[maxn],vis[2][maxn];
void curSearch(int cur){//cur代表行
    if(cur == n) tot++;
    else
    for(int i = 0;i < n;i++){//i代表列
        int ok = 1;
        c[cur] = i;
        for(int j = 0;j < cur;j++){
            if(c[cur] == c[j] || cur-c[cur]==j-c[j] || cur+c[cur]==j+c[j]){
                ok = 0;
                break;
            }
        }
        if(ok) curSearch(cur+1);
    }
}
//优化后的算法
void up_curSearch(int cur){//cur代表行
    if(cur == n) tot++;
    else
    for(int i = 0;i < n;i++){//i代表列
        if(!vis[0][i] && !vis[2][cur-i+n] && !vis[1][cur+i] ){//+n是为了把所有cur-i变为正数
            c[cur] = i;
            vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 1;
            curSearch(cur+1);
            vis[0][i] = vis[1][cur+i] = vis[2][cur-i+n] = 0;//这一步不能丢
        }
    }
}
int main()
{
    cin>>n;
    memset(vis,0,sizeof(vis));
    tot = 0;
    curSearch(0);
    //up_curSearch(0);
    cout<<tot<<endl;
    return 0;
}

最后结果是92种。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值