百练 2574(八皇后DFS)

dfs经典问题,八皇后,经典中的经典,白书里面有介绍,用到了回溯法,


规则是八个皇后不能相互攻击,当两个皇后在同一列或者同一行或者在同一个对角线上时两个皇后便会相互攻击,


所以呢用到了一个visit数组来标记并用它来判断是否在同一列或者在主对角线

原题地址:http://poj.grids.cn/

#include<stdio.h>
#include<stdlib.h>
#include<string.h> 
int tot=0;
int c[10],a[100];
int visit[3][20],ans[100][10];
void dfs(int cur)//cur表示行数,在这里面是从1开始的 
{
	if(cur==9)
	{
		++tot;//可以表示能放置成功八皇后的个数 
		for(int i=1;i<cur;i++)
		{
			ans[tot][i]=c[i];//每一次放置成功后就把这个序列存到一个二维数组中去 
		}
	}
	else
	{
		for(int i=1;i<=8;i++)
		{
			if(!visit[0][i]&&!visit[1][i+cur]&&!visit[2][i-cur+8])//visit[0][i]表示第i列已经被放置了皇后,visit[1][i+cur]表示副对角线上已经放置了皇后,visit[2][i-cur+8]表示主对角线上已经放置了皇后 ,之所以加上8是因为防止负数出现。 
			{
				c[cur]=i;//表示第cur行的皇后在第i列 
				visit[0][i]=visit[1][i+cur]=visit[2][i-cur+8]=1;
				dfs(cur+1);
				visit[0][i]=visit[1][i+cur]=visit[2][i-cur+8]=0;//当回溯时需要进行修改标记,及时回复原状 
			}
		}
	}
}
int main()
{
	int n,i,j;
	scanf("%d",&n);
	for(i=0;i<n;i++)
		scanf("%d",&a[i]);
	memset(visit,0,sizeof(visit));
    dfs(1);
    for(int k=0;k<n;k++)
    {
    	int m=a[k];
    	for(j=1;j<=8;j++)
    		printf("%d",ans[m][j]);
    	if(k!=n-1)	
    		printf("\n");	
    }
}

你可能感兴趣的:(百练 2574(八皇后DFS))