基礎練習 2n皇后問題

  基礎練習 2n皇后問題  

時間限制：1.0s   內存限制：512.0MB

      

問題描述

　　給定一個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<bits/stdc++.h>
  
int number;  
int arr[9][9];  
int bQueen[9];  
int wQueen[9];  
int count=0;  

int BlackQueen(int line)  
{  
    int temp,i;  
    for(i=0;i<line-1;i++)  
    {  
        temp=bQueen[i]-bQueen[line-1];  
        if(temp==0||temp==line-1-i||-temp==line-1-i)  
            return 0;       
	//0 == judge代表在同一列,judge==pos-1-i代表在斜率爲1的直線上,-judge == pos - 1 - i代表在斜率爲-1的直線上   
    }     
    if(line==number)  //皇后的個數等於列數  
    {  
        count++;  
        return 0;  
    } 
	 
    for(i=0;i<number;i++)            //判斷每一列   
    {  
        if(arr[line][i]&&wQueen[line]!=i)  
        {  
            bQueen[line]=i;  
            BlackQueen(line+1);  
        }  
    }  
}
  
int WhiteQueen(int line)  
{  
    int temp,i;  
    
    for(i=0;i<line-1;i++)  
    {  
        temp=wQueen[i]-wQueen[line-1];  
        if(temp==0||temp==line-1-i||-temp==line-1-i)  
            return 0;  
    }  
    
    if(line==number)            //皇后的個數等於列數  
    {  
        BlackQueen(0);          //判斷第0行  
        return 0;  
    } 
    
    for(i=0;i<number;i++)            //判斷每一列   
    {  
        if(arr[line][i])  
        {  
            wQueen[line]=i;  
            WhiteQueen(line+1);  
        }  
    }  
} 
 
int main()  
{  
    int i,j;  
    scanf("%d",&number);  
    for(i=0;i<number;i++)  
    {  
        for(j=0;j<number;j++)  
        {  
            scanf("%d",&arr[i][j]);  
        }  
    }  
    WhiteQueen(0);          //判斷第0行  
    printf("%d\n",count);  
    return 0;  
}