2n皇后问题

//2n皇后问题：1、放黑皇后，同时判断一些位置本身是否可放
//			  2、放白皇后，同时判断一些位置本身是否可放，并要判断一些位置是否已经放了黑皇后
#include <iostream>
using namespace std;
void search_1(int step);//放置黑皇后
void search_2(int step);//放置白皇后
const int maxn=100;
int n,ans=0;
int chess[maxn][maxn];//棋盘
int vis[3][maxn];//黑皇后的标记
int vis1[3][maxn];//白皇后的标记
int main()
{
	cin>>n;
	int i,j;
	for (i=0;i<n;i++)
		for (j=0;j<n;j++)
			cin>>chess[i][j];
	search_1(0);
	cout<<ans<<endl;
	return 0;
}

void search_1(int step)//(step,i)
{
	int i,j;
	if (step==n) search_2(0);//当黑皇后放完时，开始放白皇后
	else for (i=0;i<n;i++)
	{
		if (chess[step][i])//先判断该位置能不能放棋
			if (!vis[0][i] && !vis[1][step+i] && !vis[2][step-i+n])//判断 列，主对角线，副对角线
			{
				vis[0][i]=vis[1][step+i]=vis[2][step-i+n]=1;
				chess[step][i]=0;//(step,i)放入皇后
				search_1(step+1);
				vis[0][i]=vis[1][step+i]=vis[2][step-i+n]=0;//递归完后要恢复
				chess[step][i]=1;//递归完后要恢复
			}
	}
}

void search_2(int step)
{
	int i,j;
	if (step==n) ans++;//白皇后都放完后，完成一种方法。
	else for (i=0;i<n;i++)
	{
		if (chess[step][i])
			if (!vis1[0][i] && !vis1[1][step+i] && !vis1[2][step-i+n])
			{
				vis1[0][i]=vis1[1][step+i]=vis1[2][step-i+n]=1;
				chess[step][i]=0;//(step,i)放入皇后
				search_2(step+1);
				vis1[0][i]=vis1[1][step+i]=vis1[2][step-i+n]=0;
				chess[step][i]=1;
			}
	}
}