[USACO1.5]八皇后 Checker Challenge

题目描述

一个如下的 6 \times 66×6 的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列有且只有一个，每条对角线（包括两条主对角线的所有平行线）上至多有一个棋子。

上面的布局可以用序列 2\ 4\ 6\ 1\ 3\ 52 4 6 1 3 5 来描述，第 ii 个数字表示在第 ii 行的相应位置有一个棋子，如下：

行号 1\ 2\ 3\ 4\ 5\ 61 2 3 4 5 6

列号 2\ 4\ 6\ 1\ 3\ 52 4 6 1 3 5

这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出，解按字典顺序排列。
请输出前 33 个解。最后一行是解的总个数。

输入格式

一行一个正整数 nn，表示棋盘是 n \times nn×n 大小的。

输出格式

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

输入输出样例

输入 #1复制

6

输出 #1复制

2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

说明/提示

【数据范围】
对于 100\%100% 的数据，6 \le n \le 136≤n≤13。

题目翻译来自NOCOW。

USACO Training Section 1.5

 

1.这是一个dfs问题。

2.最先开始我想的是暴力加dfs，我到一个点我就判断它的行列对角线有没有queen，最终代码如下：最后一个没有AC,时间超限，但是无论我怎么优化都解决不了这个问题。

#include<stdio.h>
#define N 14
int a[N][N],b[N],count=0;
int pd(int x,int y,int n)
{
	int i,j,t=0;
	for(i=1;i<=n;i++)
		if(a[i][y]==1) t++;
	if(t!=0) return 0;
	for(i=x-1,j=y-1;i>0&&j>0&&i<=n&&j<=n;i--,j--)
		if(a[i][j]==1) t++;
	if(t!=0) return 0;
	for(i=x+1,j=y+1;i<=n&&j<=n&&i>0&&j>0;i++,j++)
		if(a[i][j]==1) t++;
	if(t!=0) return 0;
	for(i=x-1,j=y+1;i>0&&j>0&&i<=n&&j<=n;i--,j++)
		if(a[i][j]==1) t++;
	if(t!=0) return 0;
	for(i=x+1,j=y-1;i>0&&j>0&&i<=n&&j<=n;i++,j--)
		if(a[i][j]==1) t++;
	if(t!=0) return 0;
	return 1;
}
int dfs(int step,int n)//行
{
	if(step==n+1)
	{
		count++;
		if(count<=3)
		{
			for(int i=1;i<=n;i++)
				printf("%d ",b[i]);
			printf("\n");
		}
	}
	int i;
	for(i=1;i<=n;i++)
	{
		if(pd(step,i,n)==1)
		{
			a[step][i]=1;
			b[step]=i;
			dfs(step+1,n);
			a[step][i]=0;
			b[step]=0;
		}
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	dfs(1,n);
	printf("%d\n",count);
}

3.后面看了别人的题解，发现他们是直接标记的，标记行列，以及对角线。我想自己找规律于是有了下面这张图： 

 然后是这样子：

对角线如何处理？就在我第9行写的那样：往下的对角线是i-j+n,往上的是i+j-1,刚好所有对角线都可以被处理。

 所以有了如下的代码：

#include<stdio.h>
#define N 14
int a[N*2][4],b[N],count=0;
//行，列，对角线
int dfs(int step,int n)
{
	if(step==n+1)
	{
		count++;
		if(count<=3)
		{
			for(int i=1;i<=n;i++)
				printf("%d ",b[i]);
			printf("\n");
		}
	}
	int i;
	for(i=1;i<=n;i++)
	{
		if(a[step][0]==0&&a[i][1]==0&&a[step-i+n][2]==0&&a[step+i-1][3]==0)
		{
			a[step][0]=a[i][1]=a[step-i+n][2]=a[step+i-1][3]=1;
			b[step]=i;
			dfs(step+1,n);
			a[step][0]=a[i][1]=a[step-i+n][2]=a[step+i-1][3]=0;
			b[step]=0;
		}
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	dfs(1,n);
	printf("%d\n",count);
}

很高兴它AC过了。