P1219

[USACO1.5] 八皇后 Checker Challenge

题目描述

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

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

行号 \(1\ 2\ 3\ 4\ 5\ 6\)

列号 \(2\ 4\ 6\ 1\ 3\ 5\)

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

输入格式

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

输出格式

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

样例 #1

样例输入 #1

6

样例输出 #1

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

提示

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

题目翻译来自NOCOW。

USACO Training Section 1.5

代码如下

#include <bits/stdc++.h>
using namespace std;
const  int N=14;
bool visit[N];//被占领的行 
//x 设为 列方向
// y设为 
// x=y+b x=-y+b 
// b=x-y(主) b=x+y(副) 
// x 0 -n y=0-n
// x-y= -n n
// x+y = 0 2n 
// 用b来表示占领的对角线
// b范围为[-n,n] 
bool visitb1[2*N+1];
bool visitb2[2*N+1];

int k=3;//打印前三个 
int cnt=0;//总次数 
int n;
bool check(int x,int y){
	if(visit[x])return false;
	int b1=x-y;
	int b2=x+y;
	if(visitb1[b1+n])return false;
	if(visitb2[b2])return false;
	return true;
};
void add(int x,int y){
	int b1=x-y;
	int b2=x+y;
	visitb1[b1+n]=true;
	visitb2[b2]=true;
	visit[x]=true;
};
void del(int x,int y){
	int b1=x-y;
	int b2=x+y;
	visitb1[b1+n]=false;
	visitb2[b2]=false;
	visit[x]=false;
};
int t3[N];
void dfs(int x,int y){
	if(y==n-1){
		cnt++;
	}
	if(y==n-1&&k){
		k--;
		for(int i=0;i<n;i++){
			cout<<t3[i]+1<<" ";
		} cout<<endl;
	}else{
		for(int i=0;i<n;i++){
		if(check(i,y+1)){
			if(k) 
			t3[y+1]=i; 
			add(i,y+1); 
			dfs(i,y+1);
			del(i,y+1);
		}
		}	
	}
	
} 
int main() {
   // n x n的棋盘子
	cin>>n;

	memset(visit,0,sizeof visit); 
	memset(visitb1,0,sizeof visitb1); 
	memset(visitb2,0,sizeof visitb2);
	//从第一列的第一个元素开始选取 遍历第一列的所有元素 
	for(int i=0;i<n;i++){
		t3[0]=i;
		add(i,0);
		dfs(i,0);  
		del(i,0);
	}
	cout<<cnt;
    
	
	return 0;
}