POJ1321棋盘问题(DFS)

Description

在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放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
代码:
 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的退出条件为棋子下完和列超过棋盘列数。

题目链接:http://poj.org/problem?id=1321

posted @ 2019-07-31 16:48  远航。  阅读(164)  评论(0编辑  收藏  举报