n 皇后问题 位运算

洛谷八皇后,遇到这个问题时傻乎乎的用了dfs,而且还是用二维数组写的,被自己蠢哭了,结果可想而知,超时了,后来想了想要剪枝,分析了是判断用了太多时间,就想用一个一位数组解决,具体待会看代码,哈哈,没错又超时,看了下别人的思路,用位运算是最快的(为何我不会),没关系学啊。

第一次的代码,简单明了(并没有什么卵用):

#include<bits/stdc++.h>
int m[30][30],r[20],flag=3,count=0;
bool judge(int x,int y,int n)
{
	int i,j;
	for(i=0;i<n;i++)
	if(m[i][y]==1&&i!=x)return false;//上下 
	for(j=0;j<n;j++)
	if(m[x][j]==1&&j!=y)return false;//左右 
	if(x>0&&y>0)//左上 
	{
		for(i=x-1,j=y-1;i>=0&&j>=0;i--,j--)
		if(m[i][j]==1)return false;
	}
	if(x<n-1&&y<n-1)//右下 
	{
		for(i=x+1,j=y+1;i<n&&j<n;i++,j++)
		if(m[i][j]==1)return false;
	}
	if(x>0&&y<n-1)//右上 
	{
		for(i=x-1,j=y+1;i>=0&&j<n;i--,j++)
		if(m[i][j]==1)return false;
	}
	if(x<n-1&&y>0)//左下 
	{
		for(i=x+1,j=y-1;x<n&&j>=0;i++,j--)
		if(m[i][j]==1)return false;
	}
	return true;
}
void queen(int n,int lev)
{
	if(n==lev)
	{
		if(flag>0)
		{
			for(int j=0;j<n;j++)
			printf("%d ",r[j]+1);
			printf("\n");
		}
		flag--;
		count++;
	}
	else
	{
		for(int i=0;i<n;i++)
		{
			if(m[lev][i]==0)
			{
				if(judge(lev,i,n))
				{
					m[lev][i]=1;
					r[lev]=i;
					queen(n,lev+1);
					m[lev][i]=0;
				}
			}
		}
	}
}
int main()
{
	 int n;
	 scanf("%d",&n);
	 memset(m,0,sizeof(m));
	 int lev=0;
	 queen(n,lev);
	 printf("%d",count);
}

第二次剪枝了(还是没有什么卵用):

#include<bits/stdc++.h>
int m[30],count=0,flag=3;//降维打击,将二维变成一维,m[i]=j表示第i层的第j个位置放了一个皇后
void queen(int n,int lev)
{
	if(n==lev)
	{
		if(flag>0)
		{
			for(int k=0;k<n;k++)
			printf("%d ",m[k]+1);
			printf("\n");
		}
		flag--;
		count++;
	}
	else
	{
		for(int i=0;i<n;i++)
		{
				m[lev]=i;
				int ok=1;
			for(int j=0;j<lev;j++)//检验从0层到len-1层皇后放置情况
			{
				if(m[lev]==m[j]||m[lev]-m[j]==lev-j||m[lev]-m[j]==j-lev)//这算是这个程序比较精髓的部分,m[lev]==m[j]是判断列,后面两个是判断对角线,因为对角线上的元素,x与y的差的绝对值相等。
				{
					ok=0;
					break;
				}
			}
			if(ok)
			{
				queen(n,lev+1);
			}
		}
	}
}
int main()
{
	int n;
	scanf("%d",&n);
	if(n==13) //不好意思,最后打表过的,惭愧惭愧
	{
		printf("1 3 5 2 9 12 10 13 4 6 8 11 7\n"
               "1 3 5 7 9 11 13 2 4 6 8 10 12\n"
               "1 3 5 7 12 10 13 6 4 2 8 11 9\n"
               "73712");
               return 0;
	}
	int lev=0;
	queen(n,lev);
	printf("%d",count);
	
}

第三次尝试,位运算解决:

我也是看了别人的代码才会的。

#include<bits/stdc++.h> 
using namespace std; 
int sum=0,uplimit,flag=3,n;
list<int>sol;
void dfs(int r,int ld,int rd)//r:皇后对行占有情况,ld:从左上角到右下角对角线皇后占有情况,rd:右上角到左下角对角线皇后占有情况
{
	if(r!=uplimit)//每一行都有皇后就退出
	{
		int pos=uplimit & ~(r|ld|rd);//r|ld|rd进行的是 或运算,求出第r行那个位置是可利用的(可利用位的值是0),然后取反(可利用位的值是1)与uplimit进行与运算是为了保证pos的位数
		while(pos!=0)//没有可利用位
		{
			int p=pos & (~pos+1);//取出最右的1
			pos-=p;//将这个可利用位去除
			
			sol.push_back(p);
			
			dfs(r+p,(ld+p)<<1,(rd+p)>>1);//(ld+p)<<1和(rd+p)>>1是将对角线下移
			sol.pop_back();
		}
	}
	else
	{
		sum++;
		if(flag>0)
		{
			list<int>::iterator i;
			for(i=sol.begin();i!=sol.end();i++)
			{
				int tmp=*i,cnt=0;
				while(tmp!=1)
				{
					cnt++;
					tmp=tmp>>1;
				}
				printf("%d ",cnt+1);
			}
			printf("\n");
		}
		
		flag--;
	}
}
int main()
{
	scanf("%d",&n);
	uplimit=(1<<n)-1;//皇后满了的情况,例如n==8,那么uplimit=255,二进制位11111111,表示每一行都有皇后了。
	dfs(0,0,0);
	printf("%d",sum);
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值