您的位置:首页 > 其它

DFS hdu2553 N皇后问题

2015-07-20 17:40 225 查看
直接搜索就可以了

只是在标记数组vis上可以作文章

vis[0][i]来标记x为i的是否已放了棋子

vis[1][i]来标记y为i的是否已放了棋子

vis[2][i]来标记x+y为i的一斜行是否已放了棋子

vis[3][i]来标记x-y+n为i的是否已放了棋子

因为x-y可能为负数,所以加上一个n就不会出现负数了

最后,,因为测试样例非常多,但是N<=10,所以还需要打表

#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<functional>
#include<algorithm>

using namespace std;
typedef long long LL;
typedef pair<int, int> PII;

const int MX = 100 + 5;
const int INF = 0x3f3f3f3f;

int n,ans[MX];
int vis[4][MX];

void DFS(int x,int &ans){
if(x==n+1){
ans++;
return;
}

for(int y=1;y<=n;y++){
if(!vis[0][x]&&!vis[1][y]&&!vis[2][x-y+n]&&!vis[3][x+y]){
vis[0][x]=vis[1][y]=vis[2][x-y+n]=vis[3][x+y]=1;
DFS(x+1,ans);
vis[0][x]=vis[1][y]=vis[2][x-y+n]=vis[3][x+y]=0;
}
}
}

int main(){
for(n=1;n<=10;n++){
memset(vis,0,sizeof(vis));
int ret=0;
DFS(1,ret);
ans
=ret;
}

while(~scanf("%d",&n),n){
printf("%d\n",ans
);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: