题目:https://www.luogu.org/problemnew/show/P1219
题解
check[0][i]标记列
check[1][line+i]标记斜向上的对角线,对于同一条这个方向得对角线行列和相同
check[2][line-i+n]标记斜向下的对角线,对于同一条这个方向得对角线行列差相同但是可能是负数,所以加上n使其恒为正数
解法一:递归解法
#include<bits/stdc++.h>
using namespace std;
int ans[14],check[3][28]={0},sum=0,n;
void eq(int line)
{
if(line>n)//深搜完毕
{
sum++;
if(sum>3) return;//题目要求只输出字典序最小的三个
else
{//输出
for(int i=1;i<=n;i++) cout<<ans[i]<<' ';
cout<<endl;
return;//提速
}
}
for(int i=1;i<=n;i++)//遍历n列 ,保证字典序
{
if((!check[0][i])&&(!check[1][line+i])&&(!check[2][line-i+n]))
//分别代表已经布局的列 和两个方向的对角线
{
ans[line]=i;
check[0][i]=check[1][line+i]=check[2][line-i+n]=1;
eq(line+1);//继续向下搜索
check[0][i]=check[1][line+i]=check[2][line-i+n]=0;//恢复 回溯
}
}
}
int main()
{
cin>>n;//n代表棋盘大小
eq(1);
cout<<sum;
return 0;
}
解法二:全排列解法
#include<bits/stdc++.h>
using namespace std;
int main(){
int j,n,sum=0;
bool xs[28];
bool xx[28];
bool flag;
int num[15]={0};
cin>>n;
for(int i=1;i<=n;i++) num[i]=i;//初始化
do{
memset(xs,0,sizeof(xs));
memset(xx,0,sizeof(xx));
flag=true;
for(register int i=1;i<=n;i++){
if(!xs[i+num[i]]) xs[i+num[i]]=true;
else {
flag=false;
break;//break必须放在后面
}
if(!xx[i-num[i]+n]) xx[i-num[i]+n]=true;
else {
flag=false;
break;//break必须放在后面
}
}
if(flag==true){
sum++;
if(sum<=3){
for(j=1;j<=n;j++) cout<<num[j]<<' ';
cout<<endl;
}
}
}while(next_permutation(num+1,num+n+1));//这个函数太慢
// next_permutation(a,b) 对该数组前b-a个元素进行
cout<<sum;
return 0;
}