POJ1321棋盘问题

发表于 | 分类于

kuangbin带你飞专题

题目

POJ1321
在一个给定形状的棋盘（形状可能是不规则的）上面摆放棋子，棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列，请编程求解对于给定形状和大小的棋盘，摆放k个棋子的所有可行的摆放方案C。

Input

输入含有多组测试数据。
每组数据的第一行是两个正整数，n k，用一个空格隔开，表示了将在一个n*n的矩阵内描述棋盘，以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状：每行有n个字符，其中 # 表示棋盘区域， . 表示空白区域（数据保证不出现多余的空白行或者空白列）。

Output

对于每一组数据，给出一行输出，输出摆放的方案数目C （数据保证C<2^31）。

Sample Input

1
2
3
4
5
6
7
8
9
2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

Sample Output

1
2
2
1

思路

深搜，回溯

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#include <iostream>
#include <cstdio>
//#include <memory.h>

using namespace std;

int n, k;
char pan[10][10];
int row[10];
int col[10];
int sum;
int res;

void dfs(int sum, int a){
    if(sum == k){
        res++;
        return;
    }
    for(int i = a; i < n; i++){
        for(int j = 0; j < n; j++){
            if(pan[i][j] == '#' && !row[i] && !col[j]){
                row[i] = 1;
                col[j] = 1;
                dfs(sum+1, i+1);
                row[i] = 0;
                col[j] = 0;
            }
        }
    }
}

int main()
{
    while(scanf("%d %d", &n, &k) && n != -1){
        //memset(row, 0, sizeof(row));
        //memset(col, 0, sizeof(col));
        for(int i = 0; i < n; i++){
            row[i] = col[i] = 0;
        }
        res = 0;
        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){
                scanf(" %c", &pan[i][j]);
            }
        }

        dfs(0, 0);

        printf("%d\n", res);
    }

    return 0;
}
0%