7-1 N皇后问题 (用栈模拟递归)

在N * N的方格棋盘上,放置N个皇后,要求每个皇后不同行,不同列,不同左右对角线。 其中N不超过10。 要求:输出所有的解。 算法提示:用栈求解皇后问题。在这里插入图片描述
输入格式:

输入N
输出格式:

逐行输出每一种解,用每个皇后的位置坐标表示,每个位置坐标之后均有一个空格符,输出最后一行为空行。
输入样例:

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

6

输出样例:

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

1: (1,2) (2,4) (3,6) (4,1) (5,3) (6,5)
2: (1,3) (2,6) (3,2) (4,5) (5,1) (6,4)
3: (1,4) (2,1) (3,5) (4,2) (5,6) (6,3)
4: (1,5) (2,3) (3,1) (4,6) (5,4) (6,2)

思路:按照题意要求用栈模拟,学习栈的用法,借鉴了刘汝佳老师的方法标记哪些对角线已经被标记了
每个主对角线上的每个点x, y坐标之差都一样
每个副对角线上的每一个点x, y坐标之差都一样 防止为负数x - y + n
但还是没有回溯法快速
在各种优化之下才堪堪通过25ms的时限(10皇后)
当然如果自己手动实现一个栈也可以一定程度上得到优化

#include<bits/stdc++.h>

using namespace std;
const int N = 11;
struct node
{
	int a[100]; //记录数据
	int len;// 记录已经选好了几列
	bool vis1[3][100];
    //vis[0][] 标记主对角线 vis[1][]标记副对角线 vis[2][]标记每一列
};
int main()
{
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);//使cout,cin和printf一样
	int n;
	cin >> n;
	stack<node> q;
	for(int i = 1; i <= n; i++) //在第一行开始选
	{
		node tmp;
		tmp.a[1] = i;
		tmp.len = 1;
		memset(tmp.vis1, 0, sizeof tmp.vis1);
		tmp.vis1[0][1 + i] = tmp.vis1[1][1 - i + n] = tmp.vis1[2][i] = 1;
		q.push(tmp);
	}
	int cnt = 1;
	while(!q.empty())
	{
		node tmp = q.top();
		q.pop();
		if(tmp.len == n) // 输出
		{
			cout << cnt ++ << ":";
			for(int i = 1; i <= n; i++) printf(" (%d,%d)", i, tmp.a[i]);
			cout << endl;
			continue;
		}
		int len = tmp.len + 1;
		for(int i = 1; i <= n; i++)
		{
			node tmp1 = tmp;
			if(!tmp1.vis1[2][i] && !tmp1.vis1[0][len + i] && !tmp1.vis1[1][len - i + n]) //如果不在同一列且不在同一对角线
			{
				tmp1.vis1[0][len + i] = tmp1.vis1[1][len - i + n] = tmp1.vis1[2][i] = 1; //标记列和对角线
				tmp1.len ++;
				tmp1.a[tmp1.len] = i;
				q.push(tmp1);
			}
		}
	}
	return 0;
}


  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值