n皇后问题之回溯法(2021/3/26)

n皇后问题

#include<iostream>
#include<cmath>
using namespace std;
int resultCount=0;//记录解的个数
int n=0;//皇后个数
int result[1000]={0};
bool Place(int t){
    bool re=true;
    //判断与前面的t-1个皇后位置是否冲突
    for(int i=0;i<t;i++){
        if(result[t]==result[i]||t-i==fabs(result[t]-result[i])){
            re=false;
            break;
        }
    }
    return re;
}

void Backtrack(int t){
    if(t>=n){
        resultCount++;
        //打印解
        for(int i=0;i<n;i++){
            cout<<result[i]<<"  ";
        }
        cout<<endl;
        cout<<"_______\n";
    }else{
        //求第t个皇后的可能行的位置
        for(int i=0;i<n;i++){
            result[t]=i;
            if(Place(t)){
                Backtrack(t+1);
            }
        }
    }

}


int main(void)  {
    cout<<"输入皇后个数n\n";
    cin>>n;
    Backtrack(0);
    cout<<"解的个数:"<<resultCount<<endl;
    return 0;
}

测试样例

输入皇后个数n
5
0  2  4  1  3
_______
0  3  1  4  2
_______
1  3  0  2  4
_______
1  4  2  0  3
_______
2  0  3  1  4
_______
2  4  1  3  0
_______
3  0  2  4  1
_______
3  1  4  2  0
_______
4  1  3  0  2
_______
4  2  0  3  1
_______
解的个数:10

编译环境

g++
$g++ -o code code.cpp
$./code