1. 程式人生 > >藍橋杯訓練: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
測試資料
input 1
3
1 1 0 
1 1 1 
1 1 0 
output1
0
input2
4
1 1 1 1 
1 0 1 1 
1 1 1 1 
1 1 1 1 
output2
2
input 3
5
1 1 1 1 1 
1 0 1 1 1 
1 1 1 1 1 
1 0 1 1 1 
1 1 1 1 1 


output3
12


input4
6
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 


output4
12


input5
7
1 1 1 1 1 1 0 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 


output5

408

思路::比較直白的一種做法:將黑子擺滿後再去擺白子

程式碼:

/*
問題描述:
給定一個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
測試資料
input 1
3
1 1 0 
1 1 1 
1 1 0 
output1
0
input2
4
1 1 1 1 
1 0 1 1 
1 1 1 1 
1 1 1 1 
output2
2
input 3
5
1 1 1 1 1 
1 0 1 1 1 
1 1 1 1 1 
1 0 1 1 1 
1 1 1 1 1 

output3
12

input4
6
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 
1 1 1 1 1 1 

output4
12

input5
7
1 1 1 1 1 1 0 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 
1 1 1 1 1 1 1 

output5
408

*/
#include <stdio.h>
#include <stdlib.h>
int cb[8][8];
int n;
int sum;
int judge(int deep,int place,int kind)
{
    int i,j;
    if(cb[deep][place]!=1)//該位置不可放棋子
        return 0;
    for(i=0; i<deep; i++)
        for(j=0; j<n; j++)
        {
            if(cb[i][j]==kind)
            {
                if(j==place)
                    return 0;
                if(abs(deep-i)==abs(place-j))
                    return 0;
            }
        }
    return 1;
}
void dfs(int deep,int kind)
{
    if(deep==n&&kind==2)
        dfs(0,3);
    if(deep==n&&kind==3)
        sum++;
    int i;
    for(i=0; i<n; i++)
    {
        if(judge(deep,i,kind)==1)
        {
            cb[deep][i]=kind;
            dfs(deep+1,kind);
            cb[deep][i]=1;
        }
    }
}
int main()
{
    int i,j;
    scanf("%d",&n);
    for(i=0; i<n; i++)
        for(j=0; j<n; j++)
            scanf("%d",&cb[i][j]);
    dfs(0,2);
    printf("%d",sum);
    return 0;
}