经典八皇后问题
[USACO1.5]八皇后 Checker Challenge - 洛谷https://www.luogu.com.cn/problem/P1219
思路:按照题目要求每一行每一列只有一个皇后,所以先在第一行放一个皇后,然后再到下一行,先判断当前位置是否能放皇后,能的话就放皇后,然后依次往下深搜。将每种符合的情况记录下来数量,并且按照行枚举,输出1-n行皇后的列号。
做法:回溯+打标记
我有话说:
按照《算法分析与设计》上的解法,喜提87分,最后一个样例超时了。原因在于判断当前位置是否可以放置下一个皇后的时候,课本上的是遍历之前所有皇后的位置来判断当前这个皇后是否能放,相当于在backtrace()的for()循环里又套了一层循环(即使加上了内联函数inline这个遍历比较时间还是没少)。所以换一种思路,再每次放完一个皇后之后,就对当前皇后的所在行、列、主对角线、次对角线打上标记。
!标记当前皇后在的主对角线:用此时的x-y+n,因为主对角线上的每一个坐标x-y的值都恒定,相当于一维坐标轴上的每个数都对应着一组对角线;再加上有可能x-y<0,最少就是 1-n,为了让数组能够存的下,把这个一维数组整体偏移n,即x-y+n,同时每一组对角线的(x-y+n),与抽象出来的一维坐标轴一一对应的唯一性不改变,所以能保证得到x-y+n结果相同的值一定是在同一个主对角线上。
满分代码:
#include <bits/stdc++.h>
using namespace std;
int n;
int sum;
int y[20],x[20],z[100],c[100];//y[]记录每一个皇后所在的列,z[]记录主对角线,c[]记录次对角线
void backtrace(int t)
{
if(t>n){
if(sum<3)
{for(int i=1;i<=n;i++)
printf("%d ",x[i]);
printf("\n");
}
sum++;
return ;
}
else{
for(int i=1;i<=n;i++)//按列遍历,因为每次搜索下一个皇后认为她是在下一行搜索的
{
if((!y[i])&&(!z[t-i+n])&&(!c[i+t]))//原理:同一个主对角线上的数,(x-y)相同,同一个次对角线上的数,x+y相同
//采用打标记的思想将皇后可攻击范围记录下来,深搜完再恢复现场
{y[i]=1;//占领纵列
x[t]=i;
z[t-i+n]=1;
c[i+t]=1;
backtrace(1+t);
y[i]=0;
z[t-i+n]=0;
c[i+t]=0;
}
}
}
}
int main()
{
scanf("%d",&n);
backtrace(1);
printf("%d",sum);
}