N皇后问题

最新推荐文章于 2024-04-12 20:01:45 发布

原创最新推荐文章于 2024-04-12 20:01:45 发布 · 245 阅读

4 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#算法

该编程问题要求编写一个程序，使用深度优先搜索策略在给定大小的跳棋棋盘上寻找满足条件的布局，输出前三个解并计算总数。代码示例展示了如何通过递归实现这一过程。

时间限制：1秒内存限制：128M

题目描述

检查一个如下的6 x 6的跳棋棋盘，有六个棋子被放置在棋盘上，使得每行、每列只有一个，每条对角线上至多有一个棋子。

上面的布局可以用序列2 4 6 1 3 5来描述，第i个数字表示在第i行的相应位置有一个棋子，如下：
行号：1 2 3 4 5 6
列号：2 4 6 1 3 5
这只是跳棋放置的一个解。请编一个程序找出所有跳棋放置的解。并把它们以上面的序列方法输出。解按字典顺序排列。请输出前3个解。最后一行是解的总个数。
特别注意: 对于更大的N(棋盘大小N x N)，你的程序应当改进得更有效。不要事先计算出所有解然后只输出(或是找到一个关于它的公式），这是作弊。

输入描述

一个数字N (6 < = N < = 13) 表示棋盘是N x N大小的。

输出描述

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

样例

输入

输出

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

提示

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

#include<bits/stdc++.h>
using namespace std;
int chess[400],cnt,zs[100],ys[100],col[100];
int n;
bool vis[400];
void print() {
	for(int i=1; i<=n; i++) {
		cout<<chess[i]<<" ";
	}
	cout<<endl;
}
void dfs(int k) {
	if(k==n+1) {
		cnt++;
		if(cnt<=3){
			print ();
		}
		return;
	}
	for(int i=1; i<=n; i++) {
		if(!col[i]&&!ys[k+i]&&!zs[k-i+n]) {
			chess[k]=i;
			col[i]=1;
			ys[k+i]=1;
			zs[k-i+n]=1;
			dfs(k+1);
			col[i]=0;
			ys[k+i]=0;
			zs[k-i+n]=0;
		}
	}
}
int main() {
	cin>>n;
	dfs(1) ;
	cout<<cnt<<endl; 
	return 0;
}