AcWing 1432. 棋盘挑战
原题链接
中等
作者:
术
,
2021-01-22 08:38:52
,
所有人可见
,
阅读 319
#include <iostream>
using namespace std;
const int N=15;
int n;
int row[N],dg[N*2],udg[N*2];//这里要开两倍N!!!!
int path[N];
int ans[5][N];
int k;
void dfs(int u)
{
if(u>=n)
{
if(k<3)
for(int i=0; i<n; i++)
ans[k][i]=path[i];
k++;
return;
}
for(int i=0; i<n; i++)
{
if(!row[i]&&!dg[u+i]&&!udg[n-u+i])
{
row[i]=dg[u+i]=udg[n-u+i]=1;
path[u]=i+1;
dfs(u+1);
path[u]=0;
row[i]=dg[u+i]=udg[n-u+i]=0;
}
}
}
int main()
{
cin>>n;
dfs(0);
for(int i=0; i<3; i++)
{
for(int j=0; j<n; j++)
cout<<ans[i][j]<<" ";
cout<<endl;
}
cout<<k;
//cout << "Hello world!" << endl;
return 0;
}