題目
在一個給定形狀的棋盤(形狀可能是不規則的)上面擺放棋子,棋子沒有區別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列,請編程求解對於給定形狀和大小的棋盤,擺放k個棋子的所有可行的擺放方案C。
Input
輸入含有多組測試數據。每組數據的第一行是兩個正整數,n k,用一個空格隔開,表示了將在一個n*n的矩陣內描述棋盤,以及擺放棋子的數目。 n <= 8 , k <= n當為-1 -1時表示輸入結束。
隨后的n行描述了棋盤的形狀:每行有n個字符,其中 # 表示棋盤區域, . 表示空白區域(數據保證不出現多余的空白行或者空白列)。
Output
對於每一組數據,給出一行輸出,輸出擺放的方案數目C (數據保證C<2^31)。
Sample Input
2 1
#..#
4 4
...#
..#.
.#..
#...
-1 -1
Sample Output
2
1
思路:
- 和N皇后類似,一行一行進行遍歷,所以不必考慮行之間的關系。
- 在判斷一個位置是否可以放棋子,不僅要判斷這個位置是不是棋位,而且還要考慮這列是否已經放過。當放入棋子,則需要將這列設置為不可放。
- 用DFS時候,通過已放棋子數和是否超過行數來退出。
- 有可能某一行都不能放棋子。 具體看代碼 44
- 某一行的位置只能影響其下面的行不能影響同級或者上級。 具體看代碼 33-40
- 具體看代碼
代碼
#include<iostream>
#include<stdio.h>
using namespace std;
char chessboard[10000][10000];
int temp_chess[100]={0};
bool is_put[100] = {false};
int tol[100] = {0}; //成功次數
int sum; //每一次的成功次數
int put; //放的棋子數
int suc = 0; //次數
int n,k; //棋盤大小和棋子個數
bool check(int y,int r){
// 判斷該列是否可以放棋子
if(chessboard[r][y]=='.'||is_put[y]){
return false;
}
return true;
}
void DFS(int r){
// 如果所放棋子數和要求放的棋子數一樣,則成功次數++
if(put==k){
sum++;
return;
}
// 如果超了返回
if(r>=n){
return;
}
// i代表列數
for(int i=0;i<n;i++){
if(check(i,r)){ //判斷該列是否違規
is_put[i] = true;
temp_chess[r] = i;
put++;
DFS(r+1);
//該位置做的操作進行恢復,這個位置只能影響他下一行位置,而不能影響同一行位置
is_put[i] = false;
temp_chess[r] = 0;
put--;
}
}
//避免某一行都不能放棋
DFS(r+1);
}
int main(){
while(true){
sum = 0;
put = 0;
cin>>n>>k;
if(n<=0||k<=0){
break;
}
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
cin>>chessboard[i][j];
}
}
DFS(0);
tol[suc++] = sum ;
}
for(int i=0;i<suc;i++){
cout<<tol[i]<<endl;
}
return 0;
}
運行結果: