回溯法解八皇后问题

//回溯法求解八皇后问题
//Linruier
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
bool ok(int a[],int num)
{
if(num==0)return true;
for(int i=0;i<num-1;i++)if(a[num-1]==a[i])return false;
for(int i=0;i<num-1;i++)if(a[num-1]-a[i]==num-1-i||a[num-1]-a[i]==i-num+1)return false;
return true;
}
int main()
{
int a[8];//a数组记录每一行的皇后的列数
int cnt=0;
for(int i=0;i<8;i++)a[i]=0;
int k=0;int c[8];//c数组记录
for(int i=0;i<8;i++)c[i]=1;
while(k>=0)
{
while(c[k]<=8)
{a[k]=c[k]++;
if(k==7&&ok(a,8)){cnt++;break;}//是完整的解
else if(k<7&&ok(a,k+1)){k++;continue;}//是部分解
}
c[k]=1;//回溯之前把外围的位置选择恢复为8种
k--;//回溯
}
cout<<"8皇后问题共有"<<cnt<<"种解"<<endl;
}