基础练习 2n皇后问题

问题描述

　　给定一个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

样例输入

4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样例输出

0

这题可以说是n皇后问题的升级版，通过两个对book的标记（标记的是列）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
int book[2][15],ma[10][10];
int cnt;
int judge(int k,int num)
{
    int i;
    for(i=1;i<k;i++)
    {
        if(book[num][i]==book[num][k]||abs(i-k)==abs(book[num][i]-book[num][k]))//这是对斜线的判断，很巧妙
        {
            return 0;
        }
    }
    return 1;
}
void dfs(int k,int n)
{
    if(k>n)
    {
        cnt++;
    }
    else
    {
        for(int i=1;i<=n;i++)
        {
            book[0][k]=i;
            if(judge(k,0)==1&&ma[k][i]==1)
            {
                for(int j=1;j<=n;j++)
                {
                    if(j==i)
                        continue ;
                    book[1][k]=j;
                    if(judge(k,1)==1&&ma[k][j]==1)
                    {
                        dfs(k+1,n);
                    }
                }
            }
        }
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    memset(book,0,sizeof(book));
    memset(ma,0,sizeof(ma));
    cnt=0;
    for(int i=1;i<=t;i++)
    {
        for(int j=1;j<=t;j++)
        {
            scanf("%d",&ma[i][j]);
        }
    }
    dfs(1,t);
    printf("%d\n",cnt);
    return 0;
}