【思特奇杯·云上蓝桥-算法集训营】第2周 2n皇后

Description

给定一个n*n的棋盘，棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后

和n个白皇后，使任意的两个黑皇后都不在同一行、同一列或同一条对角线上，任意的两

个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法？n小于等于8。

Input

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

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

如果一个整数为0，表示对应的位置不可以放皇后。

Output

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

Sample Input

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

Sample Output

No.1

2

No.2

0

这个题就是皇后问题的一个变种。本质没有改变，依然是回溯法。

代码如下

#include<iostream>
#define N 10 
using namespace std; 
int n;
int map[N][N];
int black[N]={0};  
int white[N]={0}; 
int tot = 0;   
bool checkw(int cur) 
{  
    for(int i = 1; i < cur; i++)  
        if(white[i] == white[cur] || abs(i-cur) == abs(white[i]-white[cur]))  
            return false;  
    return true;  
}   
 
bool checkb( int cur)  
{  
    for(int i = 1; i < cur; i++)  
        if( black[i] == black[cur] || abs(i-cur) == abs(black[i]-black[cur]))  
            return false;  
    return true;  
}  
 
void dfs_white( int cur)  
{  
    if(cur == n+1)  
        tot++;  
    
    for( int i = 1; i <= n; i++)  
    {  
        if( black[cur] == i) 
            continue;       
        if( map[cur][i] == 0)  
            continue;  
        white[cur] = i;    
        if(checkw(cur))   
            dfs_white(cur+1);  
    }  
}  
  
void dfs_black( int cur)  
{  
    if( cur == n+1)  
    {  
        dfs_white(1);  
    }  
    for( int i = 1; i <= n; i++)  
    {  
       if(map[cur][i] == 0)  
            continue;  
        black[cur] = i;     
        if(checkb(cur))   
            dfs_black(cur+1);  
    }  
}  
int main(){     
   
   
   cin>>n;
   for( int i = 1; i <= n; i++)  
       for( int j = 1; j <= n; j++)  
           cin>>map[i][j];    

   dfs_black(1);    
   
   cout<<tot<<endl; 
   
   return 0;  
}