八皇后(回溯法)

八皇后问题是一个十分经典的回溯问题,我们将其分析一下可以发现,八皇后问题其实是一个全排列问题,行号代表排列的下标号,列号代表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种。

全部评论

相关推荐

10-28 14:42
门头沟学院 Java
watermelon1124:因为嵌入式炸了
点赞 评论 收藏
分享
像好涩一样好学:这公司我也拿过 基本明确周六加班 工资还凑活 另外下次镜头往上点儿
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务