POJ1321 棋盤問題(簡單搜尋)
阿新 • • 發佈:2018-12-26
題意:
在一個給定形狀的棋盤(形狀可能是不規則的)上面擺放棋子,棋子沒有區別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列,請程式設計求解對於給定形狀和大小的棋盤,擺放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
題解:
八皇后問題的變形,回溯法解決。
程式碼:
#include <iostream> using namespace std; int n,k,maxn,sum; char a[9][9]; int canplace(int r,int c) //判斷棋子能否放置。 { int i; if(a[r][c]=='.') return 0; for(i=r-1;i>=0;i--) if(a[i][c]=='Q') return 0; for(i=c-1;i>=0;i--) if(a[r][i]=='Q') return 0; return 1; } void dfs(int m) { if(k==sum) //當要求棋子數與放置的總數相同時,回溯。 { maxn++; return ; } if(m>=n*n) //當遍歷完整個棋盤,回溯。 return; else { int r,c; r=m/n; c=m%n; if(canplace(r,c)) { a[r][c]='Q'; sum++; dfs(m+1); a[r][c]='#'; sum--; } dfs(m+1); } } int main() { int i,j; while(cin>>n>>k) { if(n==-1&&k==-1) break; maxn=0; sum=0; for(i=0;i<n;i++) for(j=0;j<n;j++) cin>>a[i][j]; dfs(0); cout<<maxn<<endl; } return 0; }