2n皇后问题及拓展延伸

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

基于八皇后问题，要考虑黑白两种皇后位置是否重合，还要考虑该位置是否能放皇后代码如下

#include <cstdio>  
#include <cstring>  
#include <math.h>

int tot, c[10], n, num;                           
int map[10][10];  
int vis[100][10];  

void dfs(int cur)  
{  
    if(cur == n)                             
    {   
        for(int f = 0; f < n; f++)  
            vis[tot][f] = c[f]; 
        tot++;   
    }  
    else  
    {  
        for(int i = 0; i < n; i++)           
        {  
            if(map[cur][i] == 1)  
            {  
                int ok = 1;  
                c[cur] = i;                      
                for(int j = 0; j < cur; j++)     
                {  
                    if(c[cur] == c[j] || abs(cur - j) == abs(c[cur] - c[j]))  //判断是否在同一行、列、对角线（斜率）
                    {  
                        ok = 0;  
                        break;   
                    }   
                }   
                if(ok)  
                    dfs(cur + 1);  
            }   
        }   
    }  
}  

bool panduan(int a[], int b[]) //判断黑白皇后位置是否重合  
{  
    for(int i = 0; i < n; i++)  
    {  
        if(a[i] == b[i])     
            return true;  
    }     
    return false;  
}   

void Search(int x)                 
{  
    for(int q = 0; q < x; q++)  
    {  
        for(int w = q + 1; w < x; w++)  
            if(!panduan(vis[q], vis[w]))  
                num++;  
    }  
    printf("%d", 2 * num);  
}  

int main()  
{   
    int i, j;  
    scanf("%d", &n);  
    for(i = 0; i < n; i++)  
        for(j = 0; j < n; j++)  
            scanf("%d", &map[i][j]);  
    memset(c, -1, sizeof(c));  
    memset(vis, -1, sizeof(vis));  
    tot = 0;  
    num = 0;  
    dfs(0);  
    Search(tot);  
    return 0;  
}  

对于n皇后问题，不需要考虑那么多，直接深搜可得

#include <cstdio>  
#include <cstring>  
#include <math.h>

int tot, c[10], n, num;                           
int map[8][8];  

void dfs(int cur)  
{  
    if(cur == n)                             
        tot++;  
    else  
    {  
        for(int i = 0; i < n; i++)           
        {  
            if(map[cur][i] == 0)  
            {  
                int ok = 1;  
                c[cur] = i;                      
                for(int j = 0; j < cur; j++)     
                {  
                    if(c[cur] == c[j] || abs(cur - j) == abs(c[cur] - c[j]))  
                    {  
                        ok = 0;  
                        break;   
                    }   
                }   
                if(ok)  
                    dfs(cur + 1);  
            }   
        }   
    }  
}  


int main()  
{   
    int i, j;  
    scanf("%d", &n);  
    memset(c, -1, sizeof(c));  
    memset(map,0,sizeof(map));
    tot = 0;  
    num = 0;  
    dfs(0);  
    printf("%d",tot); 
    return 0;  
}