蓝桥杯C语言基础练习 2n皇后问题

题目描述

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

输入的第一行为一个整数n，表示棋盘的大小。接下来n行，每行n个0或1的整数，如果一个整数为1，表示对应的位置可以放皇后，如果一个整数为0，表示对应的位置不可以放皇后。
输出

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

No.1
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1


No.2
4
1 0 1 1
1 1 1 1

1 1 1 1
1 1 1 1
样例输出

No.1
2


No.2

0

这题参考的大佬的解题思路，使用递归。传送门：1.点击打开链接。

#include<stdio.h>
int whitechess[9],blackchess[9],chessboard[9][9];//定义棋盘，以及黑白棋子的位置坐标数组
int count=0;
int check(int chess[],int n)
{
    int i;
    for(i=0;i<n;i++)
    {
         int judge=chess[n]-chess[i];
         if(judge==0||judge==n-i||judge==i-n)
         {
             return 0;
         }
    }
    return 1;
}

void wqueen(int row,int n)
{
     int j;
     if(row==n)
     {
         count++;
     }else{
         for(j=0;j<n;j++)
         {
             if(chessboard[row][j]==1&&j!=blackchess[row])
             {
                 whitechess[row]=j;
                 if(check(whitechess,row))
                 {
                     wqueen(row+1,n);
                 }
             }
         }
     }     
}

void bqueen(int row,int n)
{
     int k;
     if(row==n)
     {
         wqueen(0,n);
     }else{
           for(k=0;k<n;k++)
           {
                 if(chessboard[row][k]==1)
                 {
                         blackchess[row]=k;
                         if(check(blackchess,row))
                         {
                             bqueen(row+1,n);
                         }
                 }
           }
     }
}

int main()
{
     int n,i,j;
     scanf("%d",&n);
     for(i=0;i<n;i++)
     {
        for(j=0;j<n;j++)
        {
             scanf("%d",&chessboard[i][j]);
        }
     }
     bqueen(0,n);
     printf("%d\n",count);
     return 0;
}