N皇后、2N皇后问题总结

N皇后问题

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

Input

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

Output

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

Sample Input

1
8
5
0

Sample Output

1
92
10

思路：

DFS回溯

代码：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=55+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int a[maxn][maxn];//n皇后问题
bool vis[maxn][maxn];
int n;
int ans;
bool check(int r,int c)
{
    rep(i,1,n)
    {
        rep(j,1,n)
        {
            if(!vis[i][j])
                continue;
            if(j==c)
                return 0;
            if(abs(r-i)==abs(c-j))
                return 0;
        }
    }
    return 1;
}
void dfs(int r)
{
    //cout<<r<<endl;
    if(r==n+1)
    {
        ans++;
        return ;
    }
    rep(i,1,n)
    {
        if(check(r,i))//考虑当前
        {
            vis[r][i]=1;
            dfs(r+1);
            vis[r][i]=0;
        }
    }
}
int ok[maxn];
int main()
{
    while(cin>>n&&n)
    {
        if(ok[n])
        {
            cout<<ok[n]<<endl;
            continue;
        }
        memset(vis,0,sizeof(vis));
        ans=0;
        dfs(1);
        ok[n]=ans;
        cout<<ans<<endl;
    }
}

2n皇后问题

时间限制: 1Sec 内存限制: 128MB 提交: 658 解决: 324

题目描述

给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

输入

输入的第一行为一个整数n，表示棋盘的大小。 

接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。 

输出

输出一个整数，表示总共有多少种放法。 

样例输入

4 

1  1  1  1 

1  1  1  1 

1  1  1  1 

1  1  1  1 

样例输出

2

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=5+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int a[maxn][maxn];//2n皇后问题
bool v1[maxn][maxn],v2[maxn][maxn];
int ans;
int n;
bool checkho(int r,int c)
{
    if(v2[r][c])
        return 0;
    rep(i,1,n)
    {
        rep(j,1,n)
        {
            if(v1[i][j])
            {
                if(c==j||abs(r-i)==abs(j-c))
                    return 0;//不合法
            }
        }
    }
    return 1;
}
bool checkhi(int r,int c)
{
    if(v1[r][c])
        return 0;
    rep(i,1,n)
    {
        rep(j,1,n)
        {
            if(v2[i][j])
            {
                if(c==j||abs(r-i)==abs(j-c))
                    return 0;//不合法
            }
        }
    }
    return 1;
}
void dfs(int r)
{
    //cout<<r<<endl;
    if(r==n+1)
    {
        ans++;
        return ;
    }
    rep(i,1,n)
    {
        //先放红、再放黑
        if(checkho(r,i)&&a[r][i])
        {
            v1[r][i]=1;
            rep(j,1,n)
            {
                if(checkhi(r,j)&&!v1[r][j]&&a[r][j])
                {
                    v2[r][j]=1;
                    dfs(r+1);
                    v2[r][j]=0;
                }
            }
            v1[r][i]=0;
        }
    }
}
int main()
{
    cin>>n;
    rep(i,1,n)
    {
        rep(j,1,n)
        {
            cin>>a[i][j];
        }
    }
    dfs(1);
    cout<<ans<<endl;