洛谷八皇后,遇到这个问题时傻乎乎的用了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);
}