题目链接:https://www.luogu.com.cn/problem/P1219
题目描述:
解题思路:
利用深度优先搜索遍历棋子每层能放的位置,放完之后将这一行和这一列以及对角线就不能放了。利用数组标记不能放的位置,遍历下一行的时候排除这个位置。
代码如下:
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int n,ans=0;
int a[maxn];
int col[maxn],row[maxn],line[maxn],rline[maxn];
void DFS(int k){
if(k==n+1){
if(ans<3){//输出前三种方案
for(int i=1;i<=n;i++){
cout<<a[i]<<" ";
}
cout<<endl;
}
ans++;//总方案数
return ;
}
for(int j=1;j<=n;j++){
//判断第k行第j个位置能不能放
if(col[j]==0&&row[j]==0&&line[j+k]==0&&rline[k-j+n]==0){
a[k]=j;//放入第k行第j列
col[j]=row[j]=1;//标记行和列
line[j+k]=rline[k-j+n]=1;//标记对角线
DFS(k+1);
//回溯的时候取消标记
col[j]=row[j]=0;
line[j+k]=rline[k-j+n]=0;
}
}
}
int main()
{
cin>>n;
DFS(1);
cout<<ans;
return 0;
}