回溯法解决八皇后问题

vis[2][]

#include <iostream>
#include<cstdio>
#include<string.h>
#define maxn 20
using namespace std;
int cnt=0;
int vis[3][maxn];
int n;
int aa[maxn];
void search_1(int cur)
{
if(cur==n)
cnt++;
else
{
for(int i=0; i<n; i++)
{
if(!vis[0][i] && !vis[1][cur+i] && !vis[2][cur-i+n])
{
aa[cur]=i;
vis[0][i] =vis[1][cur+i] =vis[2][cur-i+n]=1;
search_1(cur+1);
vis[0][i] =vis[1][cur+i] =vis[2][cur-i+n]=0;//回溯！！！
}
}
}
}
int main()
{
memset(vis,0,sizeof(vis));
while(scanf("%d",&n)!=EOF)
{
cnt=0;//每次读入一个n值，需要初始化一下计数器
search_1(0);//从cur=0开始递归
printf("%d\n",cnt);
}
return 0;
}