C语言 · 2n皇后问题

基础练习 2n皇后问题  

时间限制：1.0s   内存限制：512.0MB

      

锦囊1

搜索算法。

锦囊2

先搜索n皇后的解，在拼凑成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

/*
测试数据: 
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
*/
#include<stdio.h>
#include<string.h>
#include<math.h>
int hei[9];//黑皇后
int bai[9];//白皇后
int chessboard[9][9];//1:能放  0:不能放
int count = 0;

int check(int queen[],int n){//判断同一列或者两对角线是否已经放置了  
    for(int i=1; i<n; i++){  
        int judge = queen[i]-queen[n];  
        if(judge==0 || judge==i-n || judge==n-i){  
            return 0;  
        }  
    }  
    return 1;  
}  
void White(int line,int n){
    if(line==n+1){
        count++;
    }else{
        for(int i=1; i<=n; i++){
            if(chessboard[line][i]==1 && i!=hei[line]){
                bai[line]=i;
                if(check(bai,line)){
                    White(line+1,n);
                }
            }
        }
    }
}
int Black(int line,int n){
    if(line == n+1){
        White(1,n);
    }else{
        for(int i=1;i<=n;i++){
            if(chessboard[line][i]==1){
                hei[line]= i;
                if(check(hei,line)){
                    Black(line+1,n);//递归下一行 
                }
            }
        }
    }
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i =1; i <= n; i++)  
        for(int j =1; j <= n; j++)  
            scanf("%d",&chessboard[i][j]);
    Black(1,n);
    printf("%d",count);
    return 0;
}