P1219 [USACO1.5]八皇后 Checker Challenge
对于一条从右上到左下的对角线,其上的棋子坐标应满足x+y为一定值;
对于一条从左上到右下的对角线,其上的棋子坐标应满足x-y为一定值,为了避免负数的产生,代码中用x-y+n来储存数字
用一个a[][]数组判断该位置是否能放皇后,
a[1]代表该列,
a[2]代表右上到左下这条对角线,
a[3]代表左上到右下这条对角线
#include<iostream>
#include<stdio.h>
#include<iomanip>
#include<stack>
#include<queue>
#include<cstdlib>
#include <algorithm>
#include<string.h>
#include<math.h>
#define llu unsigned long long
using namespace std;
int n,sum=0,ans[15];
int a[4][30];//a[1]代表该列,a[2]代表右上到左下这条对角线,a[3]代表左上到右下这条对角线
void dfs(int i)
{
if(i>n)
{
sum++;
if(sum>3)return;
else for(int j=1;j<=n;j++)
cout << ans[j] << " " ;
cout << endl ;
return ;
}
for(int j=1;j<=n;j++)
{
if( !a[1][j] && (!a[2][i+j]) && (!a[3][i-j+n]))
{
a[1][j]=1;
a[2][i+j]=1;
a[3][i-j+n]=1;
ans[i]=j;
dfs(i+1);
a[1][j]=0;
a[2][i+j]=0;
a[3][i-j+n]=0;
}
}
}
int main()
{
cin >> n ;
dfs(1);
cout << sum << endl ;
return 0;
}