N皇后问题

题目就不多说了,昨天要交个作业,今天把想到的都记录一下;

递归回溯算法一:

#include<iostream>
#include<cstdlib>
#define N 100
using namespace std;
int board[N];
int n,ans = 0;
void Trial(int i,int n);
bool Judge(int i,int j) {
    if(j > n || i > n)return false;
    for(int l = 1; l < i; l++) {
        if(board[l] == j) return false;
        if(board[l] + i == board[i] + l || board[l] + l == board[i] + i) return false;
    }
    return true;
}
void print() {
    for(int i = 1; i <= n; i++) {
        cout << board[i] << " ";
    }
    cout << endl;
    ans ++;
}
void GoBack(int i,int n) {
    if(i < 1) {
        cout << n <<"皇后";
        cout << "一共有" << ans << "种解法" << endl;
        exit(0);
    };
    board[i]++;
    while(!Judge(i,board[i]) && board[i] <= n)board[i]++;
    if(board[i] <= n) Trial(i+1,n);
    else GoBack(i - 1,n);
}
void Trial(int i,int n) {
    if(i > n) print();
    else {
        for(int j = 1; j <=n; j++) {
            board[i] = j;
            if(Judge(i,j)) Trial(i+1,n);
        }
        GoBack(i - 1,n);
    }
}
    int main() {
        cin >> n;
        for(int i = 1; i <= n; i++)
            board[i] = 0;
        Trial(1,n);
        return 0;
    }

这是第一个想到的,里面有两个函数的相互递归调用,运行的时候发现当n= 9的时候就开始爆栈了,应该是递归调用太多了;改进了一下

递归算法二:

#include<iostream>
#include<cstdlib>
#define N 100
using namespace std;
int board[N];
int n,ans = 0;
bool Judge(int i,int j) {
    if(j > n || i > n)return false;
    for(int l = 1; l < i; l++) {
        if(board[l] == j) return false;
        if(board[l] + i == board[i] + l || board[l] + l == board[i] + i) return false;
    }
    return true;
}
void print() {
    for(int i = 1; i <= n; i++) {
        cout << board[i] << " ";
    }
    cout << endl;
    ans ++;
}
void Trial(int i,int n) {
    if(i > n) print();
    else {
        for(int j = 1; j <=n; j++) {
            board[i] = j;
            if(Judge(i,j)) Trial(i+1,n);
        }
        while(i > 1) {
            i--;
            board[i]++;
            while(!Judge(i,board[i]) && board[i] <= n)board[i]++;
            if(board[i] <= n) Trial(i+1,n);
        }
        cout << n <<"皇后";
        cout << "一共有" << ans << "种解法" << endl;
        exit(0);
    }
}
int main() {
    cin >> n;
    for(int i = 1; i <= n; i++)
        board[i] = 0;
    Trial(1,n);

    return 0;
}
这个就是把GoBack函数去掉了变成了一个函数的递归,结果也没找到哪去,n= 9 是可以算了,10的时候一样的爆栈,最后终究还是改成了非递归;

非递归回溯算法:

#include<iostream>
#include<cstdlib>
#include<time.h>
#define N 100
using namespace std;
int board[N];//存储第n行的棋子放在第几列
int n;
long long ans;
time_t Tstart,Tend;

bool Judge(int i,int j) {//判断i,j处放棋子是否合法
    if(j > n || i > n)return false;
    for(int l = 1; l < i; l++) {
        if(board[l] == j) return false;
        if(board[l] + i == board[i] + l || board[l] + l == board[i] + i) return false;
    }
    return true;
}
void print() {//输出棋子分布
    for(int i = 1; i <= n; i++) {
        cout << board[i] << " ";
    }
    cout << endl;
    ans ++;
}

void Trial(int i,int n) {//非递归回溯搜索
    int j = 1;
    while(i <= n+1) {//寻找当前第i行的位置
        if(j > n || i > n ) {//当行或者列超出棋盘时说明上一个棋子应该重新变换位置
            if(i > n ) print();//当行超出棋盘时说明当前已经产生一种输出
            while(i > 1) {//回溯
                i--;
                board[i]++;//改变位置
                while(!Judge(i,board[i]) && board[i] <= n)board[i]++;//循环寻找下一个可行位置直到找到了或者全部找完
                if(board[i] <= n) {//如果找到了则退出循环
                    i++;
                    break;;//此处break会跳到最外层while循环里
                }
            }
            if(i == 1) return;//经过while循环回溯后若当i=1的时候说明解已经全部找完了
        }
        if(i <= n) {//寻找当前棋子位置
            for( j = 1; j <=n; j++) {
                board[i] = j;
                if(Judge(i,j)) {//如果当前位置可行则寻找下一个位置
                    i++;
                    break;//此处break会跳到最外层while循环里
                }
            }
        }
    }
}
int main() {
    ios::sync_with_stdio(false);        //关闭输出流同步
    cout << "请输入皇后个数:" << endl;
    while(cin >> n) {
        Tstart = time(NULL);
        ans = 0;
        for(int i = 1; i <= n; i++)
            board[i] = 0;
        Trial(1,n);
        cout << n <<"皇后";
        cout << "一共有" << ans << "种解法" << endl;
        Tend = time(NULL);
        cout << "用时:" << Tend - Tstart  << "秒" << endl;
        cout << "请输入皇后个数:" << endl;
    }
    return 0;
}
改成非递归后就不会有爆栈的情况发生 了,n的值只要不超过棋盘范围或者结果不溢出longlong就能算,就是时间有点慢,n>15的时候就要等很久了;

另外把输出流同步关了可以省掉很多时间;

递归变非递归真的难,纠结了好久逻辑都乱了T.T。
----------------------------------------------日期分割线--------------------------------------------------

刚刚在做数独的题目的时候回头看了一下最后这个代码发现关键代码中居然有部分代码功能重复了,我就说这个题怎么这么会复杂。。。

到底还是自己先想清楚再写好啊。。附上修改后的Trial()函数

void Trial(int i,int n) {//非递归回溯搜索
    int j = 1;
    while(i <= n+1 && i > 0) {//寻找当前第i行的位置,当i= 0时说明全部解搜索完毕
            if(i > n ) {print();i--;}//当行超出棋盘时说明当前已经产生一种输出,输出当前解后回溯继续产生下一个解
            if(i > 0) {
                board[i] ++;
                while(!Judge(i,board[i]) && board[i] <= n)board[i]++;//循环寻找下一个可行位置直到找到了或者全部找完
                if(board[i] <= n) {//如果找到了则退出循环
                        //cout << i<<" --" <<   board[i] << endl;
                    i++;
                }else{//回溯
                    board[i] = 0;//回退的时候记得将当前点状态复原
                    i--;
                }
            }
    }
}
代码一下就少了十几行啊!!果然逻辑混乱是不出好代码的。-_-|||





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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值