八皇后 位运算dfs

5 篇文章 0 订阅

题目就不用描述了吧。今晚看了位操作,matrix67写的专辑 http://www.matrix67.com/blog/archives/122 里提到这个题。(今天讲搜索的时候也提到这个题 = =)

就按这个方法写了一下,其实就是dfs,思路一致,转移很直观 shr(>>) shl(<<),更新每行禁点。

速度相当快。而且很方便改成n皇后求解。

//8皇后
//92
//TIME = 0.0010s
#include<cstdio>
#include<cstring>
#include<iostream>
#include<ctime>
using namespace std;
#define LOWBIT(x) x&(-x)
/*
int t2t(int x)
{
	int ans=0;
	for(int i=0; i<8; i++)
	{
		if((x&(1<<7))>>7) ans = ans*10+1;
		else ans*=10;
		x<<=1;
	}
	return ans;
}
*/
int sum;
void solve(int c, int lc, int rc)
{
	if(c == 0xFF)	{ sum++; return; }
//	int pos = (c | lc | rc)^0xFF;		//可放		//①错误,因为lc可能会超出0xFF,导致pos也超出,导致死循环
	int pos = ((c | lc | rc)&0xFF)^0xFF;		//可放	或 0xFF&~(c|lc|rc)
	while(pos)
	{
		int p = LOWBIT(pos);
		/*		DEBUG
		printf("c=\t%08d\n", t2t(c));
		printf("lc=\t%08d\n", t2t(lc));
		printf("rc=\t%08d\n", t2t(rc));
		printf("总的=\t%08d\n", t2t((c | lc | rc)^0xFF));
		printf("当前p=\t%08d\n", t2t(p));
		printf("可放=\t%08d\n", t2t(pos&(~p)));
		*/
		solve(c|p, (lc|p)<<1, (rc|p)>>1);
		pos = pos&(~p);	//除掉可放	pos-=p;
	}
}

//	把0xFF 改成 (1<<n)-1	可解n皇后
int main()
{
	sum=0;
	solve(0, 0, 0);
	printf("%d\n", sum);
	printf("TIME = %.4lfs", (double)clock()/CLOCKS_PER_SEC);
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
八皇后问题是一个经典的回溯算法问题,但是使用位运算可以使得算法更为高效。具体来说,我们可以使用一个整数变量来表示每一行中哪些位置已经被占用,然后通过位运算来判断某个位置是否可以放置皇后。 具体的算法步骤如下: 1. 定义一个长度为8的数组board,用于存储每一行中皇后所在的列号。 2. 定义三个整数变量:col、ld、rd,分别表示列、左对角线、右对角线上已经被占用的位置。 3. 从第0行开始,依次遍历每一行。 4. 对于当前行i,遍历该行中的每一列j。 5. 判断当前位置是否可以放置皇后,即判断col、ld、rd三个变量中是否有与当前位置冲突的位置。 6. 如果当前位置可以放置皇后,则将该位置的列号存入board数组中,并更新col、ld、rd三个变量。 7. 递归处理下一行。 8. 如果已经处理完了第7行,则说明找到了一组解,输出结果。 9. 回溯到上一行,尝试其他的列。 10. 如果所有的列都尝试完了,仍然没有找到解,则回溯到上一行。 下面是使用位运算实现八皇后问题的Python代码: ```python def solveNQueens(): def dfs(row, col, ld, rd, path, res): if row == n: res.append(path) return for i in range(n): if not (col & 1 << i or ld & 1 << row + i or rd & 1 << row - i + n - 1): dfs(row + 1, col | 1 << i, ld | 1 << row + i, rd | 1 << row - i + n - 1, path + [i], res) n = 8 res = [] dfs(0, 0, 0, 0, [], res) return res print(solveNQueens()) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值