POJ-1321-棋盤問題
阿新 • • 發佈:2018-12-17
原題: 在一個給定形狀的棋盤(形狀可能是不規則的)上面擺放棋子,棋子沒有區別。要求擺放時任意的兩個棋子不能放在棋盤中的同一行或者同一列,請程式設計求解對於給定形狀和大小的棋盤,擺放k個棋子的所有可行的擺放方案C。 Input 輸入含有多組測試資料。 每組資料的第一行是兩個正整數,n k,用一個空格隔開,表示了將在一個nn的矩陣內描述棋盤,以及擺放棋子的數目。 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
#include <iostream> #include <cstdio> #include <cstring> using namespace std; int n,k,cnt,num; int dir[4][2]={1,1,1,-1,-1,-1,-1,1}; int vis[10]; char a[10][10]; void dfs(int l) { if(num==k)//如果放置的棋子等於k,則方法數量加一 { cnt++; return; } if(l>=n)//如果行數大於n就返回 return; else { for(int i=0;i<n;++i) { if(a[l][i]=='#'&&!vis[i])//如果是可以放置棋子的位置並且此行沒有被標記過 { vis[i]=1; num++; dfs(l+1); num--;//因為要計算有多少种放法,所以必須要回溯 vis[i]=0; } } } dfs(l+1);//若是當前行沒有則直接遍歷下一行 } int main() { while(scanf("%d%d",&n,&k)!=EOF) { memset(vis,0,sizeof(vis));//重置vis陣列,否則會WA num=cnt=0;//重置 if(n==-1&&k==-1)break; for(int i=0;i<n;i++) scanf("%s",&a[i]); dfs(0);//從第一行開始遍歷 printf("%d\n",cnt); } return 0; }
歡迎評論!