POJ1321棋盘问题(DFS)
Description
在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。
Input
输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。
每组数据的第一行是两个正整数,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
代码:
1 #include<iostream> 2 #include<string.h> 3 using namespace std; 4 bool row[10]; 5 char a[8][8]; 6 int ans=0,n,k; 7 bool check(int c,int i) { 8 return !row[i]&&a[i][c]=='#'; 9 } 10 void dfs(int c) { 11 if(k==0) { 12 ans++; 13 return; 14 } 15 if(c>=n) { 16 return; 17 } 18 for(int i=0; i<n; i++) { 19 if(check(c,i)) { 20 row[i]=true; 21 k--; 22 dfs(c+1); 23 row[i]=false; 24 k++; 25 } 26 } 27 dfs(c+1); 28 } 29 int main() { 30 bool flag=true; 31 while(flag) { 32 cin>>n>>k; 33 if(n==-1&&k==-1) { 34 return 0; 35 } 36 for(int i=0; i<n; i++) { 37 for(int j=0; j<n; j++) { 38 cin>>a[i][j]; 39 } 40 } 41 dfs(0); 42 cout<<ans<<endl; 43 memset(row,0,sizeof(row)); 44 ans=0; 45 } 46 }
思路分析:从列看,第一列然后行遍历,如果这列没有棋子且是棋盘的’#‘点就进一步深搜找下一列的点,dfs的退出条件为棋子下完和列超过棋盘列数。