C语言编程练习49:N皇后问题

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


Input共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。Output共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input
1
8
5
0
Sample Output
1
92
10


思路:用递归和回溯法。打表。
#include <iostream>
#include <cstdio>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <cmath>
#include <list>
#include <cstdlib>
#include <cstring>
using namespace std;
int n,sum,vis[5][50],ans[100];
void dfs(int cur)
{
    if(cur==n+1)
    {
        sum++;
        return;
    }
    for(int i=1;i<=n;i++)
    {
        if(vis[0][i]==0&&vis[1][n+i-cur]==0&&vis[2][i+cur]==0)//vis【0】表示列。vis【1】和vis【2】表示两条对角线。i表示第几列
        {
            vis[0][i]=vis[1][n+i-cur]=vis[2][i+cur]=1;//修改标记
            dfs(cur+1);
            vis[0][i]=vis[1][n+i-cur]=vis[2][i+cur]=0;//回溯
        }
    }
}

int main()
{
    for(int i=1;i<=10;i++)
    {
        memset(vis,0,sizeof(vis));
        sum=0;
        n=i;
        dfs(1);
        ans[i]=sum;
    }
    int t;
    while(cin>>t)
    {
        if(t==0)
        {
            break;
        }
        cout<<ans[t]<<endl;
    }
    return 0;
}

 代码来自灰灰考研机试班

原文地址:https://www.cnblogs.com/FantasticDoubleFish/p/14386187.html