【深度优先搜索】HDU 2553 N皇后问题

今天的主菜是搜索,想起来初中就听过八皇后问题了啊。。。当年有一个学编程的机会在我眼前我没有去珍惜,到现在我只能默默后悔。。。嘛,其实我也没那么在意啦

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2553


皇后的移动方式是直线和斜线不限长移动,毕竟最强兵种。

所以每一行必然只有一个皇后,因此我们可以选择以每行为单位来进行DFS(深度优先搜索)。

然后是竖向和斜向,竖向只要位置不同就可以了,用一个bool型数组来记录,斜向的话,只要x+y不同和x-y不同就可以保证不同行。


然后笔者就被坑了。。。


原因不明,但是肯定是出题者的阴谋。总之肯定是给了很多数据,因此我们需要预先处理一下,用一个数组保存现有的10种状态。


#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

bool side1[25],side2[25];
bool vis[11];
int n;
int ans;
int answer[20];

void Dfs(int x)
{
    if(x==n)
    {
        ans++;
        return ;
    }
    for(int i=0;i<n;i++)
    {
        if(!vis[i] && !side1[x+i] && !side2[x-i+n])
        {
            vis[i]=1;
            side1[x+i]=1;
            side2[x-i+n]=1;
            Dfs(x+1);
            vis[i]=0;
            side1[x+i]=0;
            side2[x-i+n]=0;
        }
    }
}

void init()     //预先处理
{
    for(int i=1;i<=10;i++)
    {
        n=i;
        ans=0;
        memset(vis,0,sizeof vis);
        memset(side1,0,sizeof side1);
        memset(side2,0,sizeof side2);
        Dfs(0);
        answer[i]=ans;
    }
}

int main()
{
    init();
	while(~scanf("%d",&n))
	{
        if(n==0) break;
        printf("%d\n",answer[n]);
	}
	return 0;
}



你可能感兴趣的:(搜索,ACM,DFS)