题目描述

一个如下的6×6 的跳棋棋盘,有六个棋子被放置在棋盘上,使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。 alt
上面的布局可以用序列2 4 6 1 3 5 来描述,第 i 个数字表示在第i 行的相应位置有一个棋子,如下:

行号 1 2 3 4 5 6

列号 2 4 6 1 3 5

输入格式

一行一个正整数,表示棋盘是n×n 大小的。

输出格式

前三行为前三个解,每个解的两个数字之间用一个空格隔开。第四行只有一个数字,表示解的总数。

代码部分

方法一:(全排列的优化)
#include <bits/stdc++.h>
using namespace std;
const int N=200;//数组要开的够大
int n,way[N],y[N],zd[N],fd[N],cnt;
void dfs(int u)
{
	if(u==n+1)
	{
		if(cnt<3)
		{
			for(int i=1;i<=n;i++) printf("%d ",way[i]);
			puts("");
		}
		cnt++;
		return ; 
	}
	for(int i=1;i<=n;i++)
	{
		if(!y[i]&&!zd[u+i]&&!fd[i-u+n])
		{
			y[i]=1;
			zd[u+i]=1;
			fd[i-u+n]=1;
			way[u]=i;
			dfs(u+1);
			way[u]=0;
			y[i]=0;
			zd[u+i]=0;
			fd[i-u+n]=0;
		}
	} 
}
int main ()
{
	cin>>n;
	dfs(1);
	cout<<cnt<<endl;
	return 0;	
}
方法二:(纯爆搜)
#include <bits/stdc++.h>
using namespace std;
const int N=20;
int n,way[N],x[N],y[N],zd[N],fd[N],cnt;
void dfs(int u,int i,int s)
{
	if(s>n) return;
	if(i==n+1) i=1,u++;
	if(u==n+1)
	{
		if(s==n)
		{
			if(cnt<3)
			{
				for(int i=1;i<=n;i++) printf("%d ",way[i]);
				puts("");
			}
			cnt++;
		}
		return;
	}
	if(!x[u]&&!y[i]&&!zd[u+i]&&!fd[i-u+n])
	{
		x[u]=y[i]=zd[u+i]=fd[i-u+n]=1;
		way[u]=i;
		dfs(u,i+1,s+1);
		way[u]=0;
		x[u]=y[i]=zd[u+i]=fd[i-u+n]=0;
	}
	
	dfs(u,i+1,s);
}
int main ()
{
	cin>>n;
	dfs(1,1,0);
	cout<<cnt<<endl;
	return 0;	
}
  • 可惜只能过60%的样例,我真的太菜了,那位大佬有其他想法,希望可以给我指点一二! 跪谢!