POJ 1321 棋盘问题 (DFS + 回溯)

 1 #include <iostream>
 2 #include <cmath>
 3 #include <cstdio>
 4 #include <cstring>
 5 #include <cstdlib>
 6 #include <stack>
 7 #include <queue>
 8 #include <vector>
 9 #include <algorithm>
10 #include <string>
11 
12 typedef long long LL;
13 using namespace std;
14 const int MAXN = 8;
15 int map[MAXN + 3][MAXN + 3];
16 int row[MAXN + 3];
17 int chose[MAXN + 3];
18 int n, k;
19 int ans;
20 
21 void chessDFS(int t) {
22     if(t > k) {
23         ++ans;
24     }
25     else {
26         int cnt = 0;
27         for(int i = 1; i <= n; i++) {//从1 ～ n 行找到放置第t个棋子所在的行。
28             if(chose[i]) ++cnt;
29             if(cnt == t && chose[i]) {
30                 for(int j = 1; j <= n; j++) {//在该行寻找“#”
31                     if(map[i][j] == ‘#‘ && row[j] == 0) {
32                         chose[i] = j; // 第t个棋子放在第 i行的第j个位置
33                         row[j] = 1;
34                         chessDFS(t + 1);
35                         row[j] = 0;
36                     }
37                 }
38             }
39         }
40     }
41 }
42 
43 int check(int k) {
44     for(int i = 1; i <= n; i++) {
45         if(map[k][i] == ‘#‘ && chose[k]) return 1;
46     }
47     return 0;
48 }
49 
50 void LineDFS(int t){
51     if(t > n){
52         int cnt = 0;
53         for(int i = 1; i <= n; i++) {
54             if(check(i)) cnt++;//判断chose数组中 1 的个数， 同时判断被选中行是否含有“#”，同时满足才算正真的选中
55         }
56         if(cnt == k) {//说明有K行被选中，且符合条件
57             memset(row, 0, sizeof(row));
58             chessDFS(1);//在这已经被选中的K行中放置K个棋子
59         }
60     }
61     else {
62         for(int i = 0; i < 2; i++) {//每行有两种状态，选和不选
63             chose[t] = i; //chose[t] = 1代表第t行被选则
64             LineDFS(t + 1);
65         }
66     }
67 }
68 
69 int main() {
70     while(scanf("%d%d", &n, &k) == 2 && (n != -1 && k != -1)) {
71         memset(map, ‘!‘, sizeof(map));
72         memset(row, 0, sizeof(row));
73         for(int i = 1; i <= n; i++) {
74             getchar();
75             for(int j = 1; j <= n; j++) {
76                 map[i][j] = getchar();
77             }
78         }
79         ans = 0;
80         LineDFS(1);//从N行中选择出来K行
81         printf("%d\n", ans);
82     }
83     return 0;
84 }