洛谷原题:[USACO1.5] 八皇后 Checker Challenge - 洛谷
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+1;
int a[N],b[N],c[N],d[N];//分别代表行,列,左下到右上,左上到右下
int n;
int sum=0;//解的个数
void dfs(int num)//表示从num行开始递归
{
if(num>n)
{
if(sum<=2)//控制为输出三个解
{
for(int i=1;i<=n;i++)
cout<<a[i]<<" ";
cout<<endl;
}sum++;//解的总数
return;
}
else{
for(int i=1;i<=n;i++)
{
if(b[i]==0&&c[i+num]==0&&d[i-num+n]==0)//未标记,
//这有一个可以自己推的结论:对于同一斜边,左下到右上,行列和相等。左上到右下差相等。
//所以i-num+n是为了让其始终为正值
{
a[num]=i;b[i]=1;c[i+num]=1;d[i-num+n]=1;//标记,并存储第num行的解
dfs(num+1);//开始递归下一行
b[i]=0;c[i+num]=0;d[i-num+n]=0;//刷新标记
}
}
}
}
int main(){
cin>>n;
dfs(1);
cout<<sum;
return 0;
}