码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 2553 N皇后问题 (回溯法)

时间:2014-11-20 12:00:27      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:acm   搜索   

题意  中文n*n的棋盘放n个皇后(攻击同行/列/主副对角线)  使任何两个都不互相攻击  有多少种方法

枚举每一行  用vis[3][i]记录列 主对角线 副对角线是否被占  同列和对角线都没被占就继续枚举下一行  当枚举到n+1行的时候就是一个合法答案了 

注: n*n的方阵中主对角线可以用(i-j+n)标号  副对角线可以用(i+j)标号

//ans[]={0,1,0,0,2,10,4,40,92,352,724};
#include<cstdio>
#include<cstring>
using namespace std;
const int N = 50;
int n, cnt, vis[3][N], ans[N];

void getans(int cur)
{
    if(cur == n)  ++cnt;

    for(int i = 0; cur < n && i < n; ++i)
    {
        if(vis[0][i] || vis[1][cur + i] || vis[2][cur - i + n]) continue;
        vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 1;
        getans(cur + 1);
        vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 0;
    }
}

int main()
{
    for(int i=1;i<=10;++i)
    {
        cnt = 0,n=i;
        getans(0);
        ans[i]=cnt;
    }
    while(scanf("%d", &n), n)
        printf("%d\n", ans[n]);
    return 0;
}


还有可以打印答案的

#include<cstdio>
#include<cstring>
using namespace std;
const int N = 1000;
int n, cnt, c[N], col[N][N], vis[3][N];

void getans(int cur)
{
    if(cur == n)
    {
        ++cnt;
        for(int i = 0; i < n; ++i)
            col[cnt][i] = c[i];
    }
    else
    {
        for(int i = 0; i < n; ++i)
            if(!vis[0][i] && !vis[1][cur + i] && !vis[2][cur - i + n])
            {
                c[cur] = i;
                vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 1;
                getans(cur + 1);
                vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 0;
            }
    }
}

int main()
{
    while(scanf("%d", &n))
    {
        memset(vis, 0, sizeof(vis));
        cnt = 0;
        getans(0);
        printf("%d皇后问题共有%d组解:\n\n", n, cnt);

        for(int k = 1; k <= cnt; ++k)
        {
            for(int i = 0; i < n; ++i)
            {
                for(int j = 0; j < n; ++j)
                    printf("%d ", j == col[k][i]);
                printf("\n");
            }
            printf("\n");
        }
    }
    return 0;
}

N皇后问题


Problem Description
在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。

 

Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
 

Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
 

Sample Input
1 8 5 0
 

Sample Output
1 92 10
 



HDU 2553 N皇后问题 (回溯法)

标签:acm   搜索   

原文地址:http://blog.csdn.net/acvay/article/details/41309351

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!