P1219 [USACO1.5]八皇后 Checker Challenge（DFS）C++

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;
}