特定条件的八皇后问题（回溯法）

题目

在国际象棋中，皇后是最厉害的棋子，可以横走、直走，还可以斜走。棋手马克斯·贝瑟尔 1848 年提出著名的八皇后问题：即在 8 × 8 的棋盘上摆放八个皇后，使其不能互相攻击 —— 即任意两个皇后都不能处于同一行、同一列或同一条斜线上。 要求第一行的皇后放在指定列，问有多少种摆法？

输入格式:
第一行的皇后放的列

输出格式:
解的个数

输入样例:
在这里给出一组输入。例如：

1

输出样例:
在这里给出相应的输出。例如：

4

答案

#include
#include
using namespace std;
int a[8];
int check(int k)
{
     
	for(int i=0;i<k;i++)
	{
     
		if(a[i]==a[k]||abs(i-k)==abs(a[i]-a[k]))
		return 1;
	}
	return 0;
}

int main()
{
     
	int num,cnt=0,k=1,n=8;
	cin>>num;
	fill(a,a+8,0);
	a[0]=num;
	while(k>=1)
	{
     
		a[k]++;
		while(a[k]<=8&&check(k)) a[k]++;
		if(a[k]<=n&&k==n-1)
		{
     
			cnt++;
		}
		else if(a[k]<=n&&k<n-1) k++;
		else a[k--]=0;
	}
	cout<<cnt<<endl;
}