适用于n皇后问题,经典dfs问题。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#define N 20
int c[N];//每一行的皇后放到那一列
int visit[3][N];//放皇后位置在二维数组中的行和列的关系(条件限制)
int answer=0;//放置皇后棋盘的总共方法
void dfs(int n,int cur)
{
int i,j;
if(cur==n)//结束条件,cur遍历到最后一行
{
answer++;
printf("No. %d\n",answer);
for(i=0; i<n; ++i)
{
for(j=0; j<n; ++j)
{
if(j==c[i]) printf("1 ");
else printf("0 ");
}
printf("\n");
}
return;
}
for(i=0; i<n; ++i) //遍历每一列
{
if(!visit[0][i]&&!visit[1][cur+i]&&!visit[2][cur-i+n])//检查该列是否可以放皇后(该位置是否被占用,副对角线占用情况,主对角线占用情况)
{
c[cur]=i;
visit[0][i]=visit[1][cur+i]=visit[2][cur-i+n]=1;
dfs(n,cur+1);
visit[0][i]=visit[1][cur+i]=visit[2][cur-i+n]=0;
}
}
}
int main()
{
memset(visit,0,sizeof(visit));
dfs(8,0);//可以改成任意皇后问题
printf("%d\n",answer);
return 0;
}