完整程序:
#include<bits/stdc++.h>
using namespace std;
//n皇后問題回溯算法的實現
//(1) n皇后問題回溯算法的數據結構
#define NUM 20
int n; //棋盤的大小
int x[NUM]; //解向量
int sum; //當前已經找到的可行方案數
//(3) 檢查當前皇后位置的約束函數
//形參t是回溯的深度
inline bool Place(int t)
{
int i;
for(i=1;i<t;i++)
if((abs(t-i)==abs(x[i]-x[t]))||(x[i]==x[t]))
return false;
return true;
}
//(2) n皇后問題回溯算法的實現
//形參t是回溯的深度,從1開始
void Backtrack(int t)
{
int i;
//到達葉子結點,獲得一個可行方案。累計總數,並輸出該方案
if(t>n)
{
sum++; //是全局變量
for(i=1;i<=n;i++)
printf("%d",x[i]);
printf("\n");
}
else
for(i=1;i<=n;i++)
{
x[t]=i;
if(Place(t))Backtrack(t+1);
}
}
int main(){
cout<<"請輸入n皇后的規模:"<<endl;
cin>>n;
cout<<n<<"x"<<n<<"格的棋盤上放置"<<n<<"個皇后並使其不能互相攻擊的所有方案爲:"<<endl;
Backtrack(1);
cout<<"着色方案總數爲"<<endl;
cout<<"Total="<<sum;
return 0;
}